今天在刷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()函数的时候,一定一定要注意两个切片的长度一定要相同!!!