Golang闭包引起的Slice传参问题

原创已于 2023-06-14 07:09:34 修改 · 216 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#golang #leetcode #算法

于 2023-06-13 13:38:40 首次发布

算法专栏收录该内容

1 篇文章

订阅专栏

在LeetCode使用Golang解题时遇到闭包导致的Slice参数问题。递归过程中，由于Go的闭包机制，传递的Slice在函数调用中被意外修改。解决方案是通过返回和接收Slice来保持其正确状态，从而实现预期的功能。

最近在leetcode用golang刷题的时候发现golang闭包的一个坑。

原题：给你二叉树的根节点 root ，返回其节点值的 层序遍历。

我琢磨着能否用递归的方式解决，于是有了以下代码：

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}
//主方法
func levelOrder(root *TreeNode) [][]int {
	var arr = [][]int{}
	traverse(root, 0, arr)
	return arr
}

//递归方法
func traverse(node *TreeNode, level int, arr [][]int) {
	if len(arr) < level+1 {
		arr = append(arr, []int{})
	}
	if node == nil {
		return
	} else {
		arr[level] = append(arr[level], node.Val)
	}
	traverse(node.Left, level+1, arr)
	traverse(node.Right, level+1, arr)
}

//初始化Tree
func initTree() *TreeNode {
	node5 := TreeNode{
		Val: 7,
	}
	node4 := TreeNode{
		Val: 3,
	}
	node3 := TreeNode{
		Val:   6,
		Left:  &node4,
		Right: &node5,
	}
	node2 := TreeNode{
		Val: 4,
	}
	node1 := TreeNode{
		Val:   5,
		Left:  &node2,
		Right: &node3,
	}
	return &node1
}
//打印结果数组
func printArr(arr [][]int) {
	count := len(arr)
	for i := 0; i < count; i++ {
		for j, _ := range arr[i] {
			fmt.Printf("\t %d", arr[i][j])
		}
	}
}
//调用代码
func main() {
	root := initTree()
	arr := levelOrder(root)
	printArr(arr)
}

看着好像没啥问题，但是一运行，打印出的结果是空的。

Debug的时候发现传递的arr数组的值随着调用方法的往复，数组的内容也不断变化，等到主函数的时候arr直接空了。于是心里想，这不是js里面的闭包么？于是网上搜了下，发现go真有闭包机制。难怪不能像java/C#一样一路把slice传到底。

最终，想了个解决方案：

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}
//主函数
func levelOrder(root *TreeNode) [][]int {
	var arr = [][]int{}
	//接住返回slice
	arr = traverse(root, 0, arr)
	return arr
}

//递归函数
//修改：返回slice
func traverse(node *TreeNode, level int, arr [][]int) [][]int {
	if len(arr) < level+1 {
		arr = append(arr, []int{})
	}
	if node == nil {
		return arr
	} else {
		arr[level] = append(arr[level], node.Val)
	}
	//这里我们接住返回的slice，并且继续返回
	arr = traverse(node.Left, level+1, arr)
	arr = traverse(node.Right, level+1, arr)
	return arr
}
//初始化树
func initTree() *TreeNode {
	node5 := TreeNode{
		Val: 7,
	}
	node4 := TreeNode{
		Val: 3,
	}
	node3 := TreeNode{
		Val:   6,
		Left:  &node4,
		Right: &node5,
	}
	node2 := TreeNode{
		Val: 4,
	}
	node1 := TreeNode{
		Val:   5,
		Left:  &node2,
		Right: &node3,
	}
	return &node1
}

//打印结果数组
func printArr(arr [][]int) {
	count := len(arr)
	for i := 0; i < count; i++ {
		for j, _ := range arr[i] {
			fmt.Printf("\t %d", arr[i][j])
		}
	}
}

func main() {
	root := initTree()
	arr := levelOrder(root)
	printArr(arr)
}

根据代码可以看出，想要把arr进行改变，我们需要不断地把arr返回并接收，最终返回到主函数。运行起来就可以得到正常的结果了