Go语言面试题：forrange中指针为何指向最后元素？

本文讲解Go语言面试中一个关于`for...range`循环与指针的常见问题。在使用`for...range`循环遍历切片时，如果将指向循环变量的指针存储到map中，由于Go语言`for...range`循环复用迭代变量的特点，最终map中所有指针都将指向最后一个元素。文章通过一个学生信息存储的例子，详细分析了这个问题产生的原因，并提供了通过创建迭代变量副本的解决方案，有效避免了指针指向同一个内存地址的问题，帮助开发者深入理解Go语言`for...range`循环的机制及指针的使用技巧。

Go语言面试题：深入理解for...range循环与指针

Go语言的for...range循环简洁高效，但使用指针时容易引发误解。本文剖析一个常见的面试题，阐明for...range循环中指针的陷阱及解决方法。

问题描述

考虑一个学生结构体和学生切片，目标是将学生信息存储到map中并打印姓名：

type student struct {
    name string
    age  int
}

func main() {
    m := make(map[string]*student)
    stus := []student{
        {name: "pprof.cn", age: 18},
        {name: "测试", age: 23},
        {name: "博客", age: 28},
    }

    for _, stu := range stus {
        m[stu.name] = &stu
    }
    for k, v := range m {
        fmt.Println(k, "=>", v.name)
    }
}

预期输出：

pprof.cn => pprof.cn
测试 => 测试
博客 => 博客

实际输出：

pprof.cn => 博客
测试 => 博客
博客 => 博客

问题分析

所有v.name都变成了"博客"，原因在于for...range循环的迭代变量stu在每次迭代中是同一个变量，只是其值被更新。&stu创建的指针始终指向这个单一变量。循环结束后，stu的值为最后一个学生的信息，因此map中的所有指针都指向了这个最终的stu变量。

Go语言的for...range循环复用迭代变量，不会为每次迭代创建新变量。因此，获取迭代变量的地址将始终得到同一个地址。

解题方法

为了解决这个问题，需要在每次迭代中创建stu的副本：

for _, stu := range stus {
    stuCopy := stu // 创建副本
    m[stu.name] = &stuCopy
}

通过创建stuCopy副本，并使用&stuCopy作为map的值，确保每个学生的信息都被独立存储和引用。

正确输出

修改后的代码将产生预期的输出：

pprof.cn => pprof.cn
测试 => 测试
博客 => 博客

总结

本例揭示了for...range循环中使用指针的潜在风险。 理解迭代变量的复用机制，并在必要时创建副本，是避免此类问题的关键。 这对于Go语言开发者来说是一个重要的知识点。

到这里，我们也就讲完了《Go语言面试题：forrange中指针为何指向最后元素？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！