113. 路径总和 II
题目描述
解题思路
1、前序遍历判断节点是否满足路径和
2、需要返回所有满足的路径和,而每条路径也需要记录,所以需要定义两个list:ret, path。ret记录结果;path记录路径,采用回溯法,因此只初始化了一次。
代码实现
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def pathSum(self, root: TreeNode, targetSum: int) -> List[List[int]]:
ret = []
path = []
def pathSum(root, targetSum):
if not root:
return
path.append(root.val)
targetSum -= root.val
if targetSum == 0 and not root.left and not root.right:
ret.append(path[:]) # path[:]重点,取list所有值
pathSum(root.left, targetSum)
pathSum(root.right, targetSum)
# 重点:本题需要回溯,那为什么只pop()一次?最简单想法是只append()了一次。
# 详细点说:本题需要遍历所有路径,假如遍历到节点node,按照程序走会先判断是否满足路径目标和;
# 然后依次遍历左右子树;当node节点处理完就需要向上回溯父节点,遍历兄弟节点。
path.pop()
pathSum(root, targetSum)
return ret
参考文献
1、官方题解