Go语言中for-range循环的陷阱：理解值拷贝与切片元素修改

package main

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

// Weight 和 Spread 是示例中使用的自定义类型
type Spread float64
type Weight struct {
    S Spread
    X float64
    Y float64
}

type Stack []Weight

func newStackIncorrect(size int, startSpread Spread) Stack {
    stack := make(Stack, size) // 创建一个包含 'size' 个零值 Weight 的切片

    for _, curWeight := range stack { // 遍历切片，curWeight 是每个元素的副本
        // 这里的赋值操作只影响了 curWeight 这个副本
        // 原始 stack 中的元素并未被修改
        curWeight = Weight{startSpread, rand.Float64(), rand.Float64()}
        fmt.Printf("Inside loop (copy): %v\n", curWeight) // 打印的是副本的值
    }

    return stack // 返回的 stack 仍然是包含零值 Weight 的切片
}

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

    initialSpread := Spread(1.0)
    incorrectStack := newStackIncorrect(3, initialSpread)
    fmt.Println("Incorrectly initialized stack:")
    for i, w := range incorrectStack {
        fmt.Printf("stack[%d]: %v\n", i, w)
    }
    // 预期输出将是 Weight 结构体的零值，因为内部的修改没有生效
}

package main

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

// Weight 和 Spread 是示例中使用的自定义类型
type Spread float64
type Weight struct {
    S Spread
    X float64
    Y float64
}

type Stack []Weight

func newStackCorrect(size int, startSpread Spread) Stack {
    stack := make(Stack, size) // 创建一个包含 'size' 个零值 Weight 的切片

    for i := 0; i < size; i++ { // 使用索引 i 遍历切片
        // 直接通过索引访问并修改 stack 中的元素
        stack[i] = Weight{startSpread, rand.Float64(), rand.Float64()}
        fmt.Printf("Inside loop (original element): stack[%d] = %v\n", i, stack[i]) // 打印的是原始元素的值
    }

    return stack // 返回的 stack 包含了正确初始化的 Weight 元素
}

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

    initialSpread := Spread(1.0)
    correctStack := newStackCorrect(3, initialSpread)
    fmt.Println("\nCorrectly initialized stack:")
    for i, w := range correctStack {
        fmt.Printf("stack[%d]: %v\n", i, w)
    }
    // 预期输出将是包含随机值的 Weight 结构体，因为内部的修改已经生效
}