leetcode | Path Sum II

题目:将返回一棵二叉树从根到叶节点所经过的值之和等于Sum的路径。

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and  sum = 22 ,
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

return

[
   [5,4,11,2],
   [5,8,4,5]
]

思路:

利用递归+深度优先搜索。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def pathSum(self, root, Sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: List[List[int]]
        """
        if not root:
            return []
        res = []
        self.excute(root,[],Sum,res)
        return res
        
    def excute(self,root,path,Sum,res):
        path.append(root.val)
        path_left = path[:] 
        path_right = path[:]

        if not root.left and not root.right: #judge if is a leaf
            if Sum-root.val==0:
                res.append(path)
            return
        
        if root.left:
            self.excute(root.left,path_left,Sum-root.val,res)
        if root.right:
            self.excute(root.right,path_right,Sum-root.val,res)

        
代码比较好理解,这里主要讲一下在 excute函数中,为什么每次递归都要拷贝一次path?如下所示:

path_left = path[:] 
path_right = path[:]
首先看下面代码:

a = [1,2,3]
b = a
a.append(4)
print b

#b:[1,2,3,4]
上述代码说明了,如果使用语句 b=a那么,b指向a,所以当修改a时,b也会随之改变。因此,如果我们不使用拷贝path的方法,最后path将储存树中的所有节点。









评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值