问题描述:
在做剑指 Offer 34. 二叉树中和为某一值的路径的时候遇见了需要进行树递归的操作,递归的过程中需要根据条件把tmp_list append 到anslist里面,但是最后的结果出现如下情况:
> 第一轮:
> tmp_list=[1,2,3]
> ans = [[1,2,3]]
> 第二轮,此时tmp_list=[4,5,6]
> 这个时候,奇怪的事情发生了, ans = [[4,5,6], [4,5,6]]
现象解释:
这个时候,发现最开始存在ans里面存的tmp_list已经变成最新的tmp_list,也就是说现在的ans里面都一样了。这个就是原因是因为每次ans.append(tmp_list)添加的是引用关系,每一个添加进去的都是tmp_list的引用,自然就都是一样的,所以我们应该每次将tmp_list拷贝一下,然后添加到ans中,这样即使后面tmp_list再有变化,也不会已经影响已经存在在ans中的值了。关于深拷贝浅拷贝可以参考这个博客list拷贝,我这里使用了deepcopy()函数进行深拷贝,解决了这个问题。代码如下:
class Solution:
def pathSum(self, root: TreeNode, target: int) -> List[List[int]]:
ans = []
path = []
def mid(root):
if not root: return False
path.append(root.val)
# print("!!!!!!!!", path)
if not root.left and not root.right:
if sum(path) == target:
tmp = deepcopy(path) # 这里非常的重要,因为如果没有使用拷贝,其实aoppend加进去的是一种引用,所以后面都会改。
ans.append(tmp)
# print("++++++", path)
# print("ansansasn", ans)
path.pop(-1)
return
mid(root.left)
mid(root.right)
path.pop(-1)
mid(root)
return ans