引言
树的求和属于树的题目中比较常见的,因为可以有几种变体,灵活度比较高,也可以考察到对于树的数据结构和递归的理解。
112. 路径总和
给你二叉树的根节点
root
和一个表示目标和的整数targetSum
,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。
叶子节点 是指没有子节点的节点。
示例:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
思路1:递归
这道题是判断是否存在从根到叶子的路径和跟给定sum相同。树的题目基本都是用递归来解决,主要考虑两个问题:
如何把问题分治成子问题给左子树和右子树。这里就是看看左子树和右子树有没有存在和等于sum 减去当前结点值的路径
,只要有一个存在,那么当前结点就存在路径。
考虑结束条件是什么︰
结束条件1:如果当前节点是空的,则返回false。
结束条件2:如果是叶子,那么如果剩余的sum等于当前叶子的值,则找到满足条件的路径,返回true。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def hasPathSum(self, root: TreeNode, sum: int) -> bool:
# 如果当前节点为空,则返回False
if root == None:
return False
# 如果当前节点是叶子节点,并且剩余sum等于叶子节点值
if root.left == None and root.right == None and root.val == sum:
return True
# 如果当前节点是非叶子节点,则对它的左右子树分别调用函数
return self.hasPathSum(root.left,sum-root.val) or self.hasPathSum(root.right,sum-root.val)
思路2:迭代
我们可以用栈将递归转成迭代的形式。
栈中保存当前节点前的剩余值就可以了。
# 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 hasPathSum(self, root: TreeNode, targetSum: int) -> bool:
# 使用栈的结构,将递归化为迭代
if root == None:
return False
# 栈
stack = [(root,targetSum)]
while stack:
node,sum = stack.pop()
# 如果为叶子节点,判断叶子节点是否满足条件
if node.left == None and node.right == None and node.val == sum:
return True
# 如果不为叶子节点,则将当前节点的孩子节点及剩余值压入栈
if node.left != None:
stack.append((node.left,sum-node.val))
if node.right != None:
stack.append((node.right,sum-node.val))
return False
113. 路径总和 II
这道题思路和112–路径总和是完全一样的,只是需要输出所有路径,所以需要数据结构来维护路径,添加两个参数,一个用来维护走到当前结点的路径,一个用来保存满足条件的所有路径,思路上递归条件和结束条件是完全一致的,空间上这里会依赖于结果的数量
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有