Go语言中for-range循环的变量声明与元素修改：避开常见陷阱

for index, value := range collection {
    // ...
}

package main

import (
    "fmt"
    "math/rand"
    "time"
)

// 示例类型定义，模拟实际应用场景
type Spread float64
type Weight struct {
    SpreadValue Spread
    Rand1       float64
    Rand2       float64
}
type Stack []Weight

// 错误示例：尝试通过for-range修改切片元素
func newStackIncorrect(size int, startSpread Spread) Stack {
    stack := make(Stack, size) // 创建一个包含 'size' 个零值 Weight 的切片

    // 循环遍历stack，curWeight是每个元素的副本
    for _, curWeight := range stack {
        // 这里的赋值操作只修改了curWeight这个副本，
        // 而stack中的原始元素保持不变。
        // Go编译器会提示：curWeight declared and not used
        curWeight = Weight{startSpread, rand.Float64(), rand.Float64()}
        // 实际上，这个赋值后的curWeight在当前迭代结束时就被丢弃了
    }
    return stack // 返回的stack仍然是零值元素填充的
}

func main() {
    rand.Seed(time.Now().UnixNano()) // 初始化随机数种子

    initialStack := newStackIncorrect(3, 10.0)
    fmt.Println("不正确方式创建的Stack:")
    for i, w := range initialStack {
        fmt.Printf("Stack[%d]: %+v\n", i, w)
    }
    // 预期输出将是 Weight 结构体的零值，例如：
    // Stack[0]: {SpreadValue:0 Rand1:0 Rand2:0}
    // Stack[1]: {SpreadValue:0 Rand1:0 Rand2:0}
    // Stack[2]: {SpreadValue:0 Rand1:0 Rand2:0}
}

package main

import (
    "fmt"
    "math/rand"
    "time"
)

// 示例类型定义
type Spread float64
type Weight struct {
    SpreadValue Spread
    Rand1       float64
    Rand2       float64
}
type Stack []Weight

// 正确示例：通过索引修改切片元素
func newStackCorrect(size int, startSpread Spread) Stack {
    stack := make(Stack, size)

    // 使用传统for循环，通过索引i直接访问并修改stack[i]
    for i := 0; i < size; i++ {
        stack[i] = Weight{startSpread, rand.Float64(), rand.Float64()}
    }
    return stack
}

// 另一种正确方式：使用带有索引的for-range循环
func newStackCorrectWithRangeIndex(size int, startSpread Spread) Stack {
    stack := make(Stack, size)

    // for-range也可以提供索引，通过索引i直接访问并修改stack[i]
    for i := range stack { // 这里我们只需要索引，所以省略了值部分
        stack[i] = Weight{startSpread, rand.Float64(), rand.Float64()}
    }
    return stack
}

func main() {
    rand.Seed(time.Now().UnixNano())

    // 使用正确方式创建Stack
    correctStack := newStackCorrect(3, 10.0)
    fmt.Println("正确方式创建的Stack:")
    for i, w := range correctStack {
        fmt.Printf("Stack[%d]: %+v\n", i, w)
    }

    fmt.Println("\n使用带有索引的for-range方式创建的Stack:")
    correctStackWithRangeIndex := newStackCorrectWithRangeIndex(3, 20.0)
    for i, w := range correctStackWithRangeIndex {
        fmt.Printf("Stack[%d]: %+v\n", i, w)
    }
}