您的位置:首页 > 编程语言 > Python开发

leetcode-路径总和II(python)

2019-03-23 20:47 190 查看

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

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

示例:

给定如下二叉树,以及目标和 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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: