leetcode-路径总和II(python)

题目:
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:

给定如下二叉树,以及目标和 sum = 22,

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1

返回:

[
[5,4,11,2],
[5,8,4,5]
]
思想:
二叉树深度遍历+回溯
代码:

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
        pathvalue=0
        path=[]
        result=[]
        def preorder(node,pathvalue,sum,path,result):
            if node==None:
                return
            pathvalue+=node.val
            path.append(node.val)
            if pathvalue==sum and node.left==None and node.right==None:
                result.append(list(path))  #注意加list
            preorder(node.left,pathvalue,sum,path,result)
            preorder(node.right,pathvalue,sum,path,result)
            pathvalue-=node.val
            path.pop()
        preorder(root,pathvalue,sum,path,result)
        return result

猜你喜欢

转载自blog.csdn.net/weixin_40876685/article/details/88768024