题目分析:
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。说明: 叶子节点是指没有子节点的节点。
示例:给定如下二叉树,以及目标和 sum = 22,
返回: [[5,4,11,2],[5,8,4,5]]
解题思路:
这一题是【LeetCode】112. Path Sum的进阶版,比上一题要稍微难一些,但是用上一题的思路可以解决。
- 如果根节点为空,直接返回[]
if not root: return []
- 如果当前节点有值,加入path
if root: path.append(root.val)
- 如果左右节点为空,且当前节点值等于sum就将path加入res
if not root.left and not root.right and root.val == sum:
res.append(list(path))
- 如果左节点有值,递归左节点,递归返回时证明不满足,要将path中最后一个节点pop
if root.left:
self.dfs(root.left, sum - root.val, path, res)
path.pop()
- 如果右节点有值,递归右节点,递归返回时证明不满足,要将path中最后一个节点pop
if root.right:
self.dfs(root.right, sum - root.val, path, res)
path.pop()
提交代码:(递归,Runtime: 40 ms, faster than 99.97% )
class Solution:
def pathSum(self, root: TreeNode, sum: int) -> 'List[List[int]]':
if not root: return []
res = []
self.dfs(root, sum, [], res)
return res
def dfs(self, root, sum, path, res):
if root: path.append(root.val)
if not root.left and not root.right and root.val == sum:
res.append(list(path))
if root.left:
self.dfs(root.left, sum - root.val, path, res)
path.pop()
if root.right:
self.dfs(root.right, sum - root.val, path, res)
path.pop()