Golang中的append的使用

今天在刷leetcode的时候,第113题让我遇到了一个Go语言中append函数的一个坑。
这道题为:https://leetcode-cn.com/problems/path-sum-ii/

我的解题答案开始是这个:

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func pathSum(root *TreeNode, sum int) [][]int {
	if root == nil {
		return make([][]int, 0, 0)
	}
	var val = root.Val
	if val == sum && root.Left == nil && root.Right == nil {
		return [][]int{[]int{root.Val}}
	}
	
	var finalRes = make([][]int, 0, 0)
	var res = []int{root.Val}
	finalRes = dfs(root.Left, val, sum, res, finalRes)
	finalRes = dfs(root.Right, val, sum, res, finalRes)

	return finalRes
}

func dfs(root *TreeNode, val int, sum int, res []int, finalRes [][]int) [][]int {
	if root == nil {
		return finalRes
	}
	val += root.Val
	res = append(res, root.Val)
	if val == sum && root.Left == nil && root.Right == nil {
		finalRes = append(finalRes, res)
		return finalRes
	}

	finalRes = dfs(root.Left, val, sum, res, finalRes)
	var x = finalRes
	finalRes = dfs(root.Right, val, sum, res, x)

	return finalRes
}

注意dfs函数中,

res = append(res, root.Val)

这一行。
当输入为:

[5,4,8,11,null,13,4,7,2,null,null,5,1]
22

时,答案为:[[5,4,11,2],[5,8,4,1]]
而正确答案应为:[[5,4,11,2],[5,8,4,5]]

但是,你看我的算法流程,逻辑没问题呀~咋就出问题了呢?

百思不得其解的我,只好在IDE里进行了调试,于是发现,append函数是个坑!!!

当我运行到叶子节点5这里的时候,我清楚的看到,res=[5,8,4,5],而且finalRes=[[5,4,11,2],[5,8,4,5]]
没一点问题,但是程序继续往下运行到叶子节点1这里的时候,准确代码就是在res=append(res, root.Val)这里的时候,神奇的一幕出现了,
finalRes居然直接从[[5,4,11,2],[5,8,4,5]]变成了[[5,4,11,2],[5,8,4,1]]!!!

也就是,我在递归函数后面,修改了res的值,而finalRes在上一个递归调用的时候,finalRes=append(finalRes, res),这个时候,finalRes就相当于和res绑在一块了,修改了res,那么finalRes也会变!!!

无奈之下,我只好另谋出路。

看了下leetcode上的题解,得知了他们用了一个copy函数,也即在对finalRes添加新值的时候,我只用一个新的值而不用res,这样,问题就解决了。

修改后的代码如下:

func pathSum(root *TreeNode, sum int) [][]int {
	if root == nil {
		return make([][]int, 0, 0)
	}
	var val = root.Val
	if val == sum && root.Left == nil && root.Right == nil {
		return [][]int{[]int{root.Val}}
	}

	var finalRes = make([][]int, 0, 0)
	var res = []int{root.Val}
	finalRes = dfs(root.Left, val, sum, res, finalRes)
	finalRes = dfs(root.Right, val, sum, res, finalRes)

	return finalRes
}

func dfs(root *TreeNode, val int, sum int, res []int, finalRes [][]int) [][]int {
	if root == nil {
		return finalRes
	}
	val += root.Val
	res = append(res, root.Val)
	if val == sum && root.Left == nil && root.Right == nil {
		var x = make([]int, len(res))
		copy(x, res)
		finalRes = append(finalRes, x)
		return finalRes
	}

	finalRes = dfs(root.Left, val, sum, res, finalRes)
	finalRes = dfs(root.Right, val, sum, res, finalRes)

	return finalRes
}

最后,完美通过。
需要注意的一点是:在使用copy()函数的时候,一定一定要注意两个切片的长度一定要相同!!!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值