题目来源:https://leetcode-cn.com/problems/path-sum-ii/
注:112与本题基本一样,稍微改一下就可以了 https://leetcode-cn.com/problems/path-sum/
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
首先,从题目的“从根节点到叶子节点”这一句,可以很自然的想到深度先序遍历,按根-左子数-右子树的顺序访问。
那么就可以采用先序遍历的形式,每次将当前节点的值记录下来,传递给当前节点的左右子树,直到叶节点,则判断哪一条路径的节点列表和为target。
解法一:非递归(比较好理解)
# 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, target):
"""
:type root: TreeNode
:type target: int
:rtype: List[List[int]]
"""
if not root:
return []
stack = [([root.val], root)] # 先序遍历,根先入栈,注意存值用list,方便后面直接对list进行sum
ans = []
while stack:
val, tmp = stack.pop()
if tmp.right:
stack.append((val + [tmp.right.val], tmp.right))
if tmp.left:
stack.append((val + [tmp.left.val], tmp.left))
if not tmp.right and not tmp.left and sum(val) == target: # 叶节点,且路径节点值的和为目标
ans.append(val) # 增加路径节点值
return ans
解法二:递归
# 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, target):
"""
:type root: TreeNode
:type target: int
:rtype: List[List[int]]
"""
ans = []
def dfs(root, target, tmp):
if not root:
return
if not root.right and not root.left and target-root.val == 0: # 叶节点,且target与路径节点值相减后为0
tmp += [root.val] # 路径节点值加上当前节点(叶节点)
ans.append(tmp) # 满足条件的路径
dfs(root.left, target - root.val, tmp + [root.val]) # 每次递归时,target减去已经访问过的节点的值,tmp加上这个值
dfs(root.right, target - root.val, tmp + [root.val])
dfs(root, target, [])
return ans
参考:
https://leetcode-cn.com/problems/path-sum-ii/solution/dfsfei-di-gui-python3-by-baiyizhe/