题目描述
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
叶子节点 是指没有子节点的节点。
实例1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
解释:等于目标和的根节点到叶节点路径如上图所示。
提示
树中节点的数目在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
思路
对于二叉树,大都和遍历有关。
本题中关键是2点:
- 计算节点的值。
- 从根节点到叶子节点的路径。
- 返回值。
图、树等题型基本是遍历问题,可以从DFS遍历入手,计算节点值即相当于教材里对当前节点的访问,值得注意的是返回值问题,针对题目含义,存在一条这样的路径即可返回true
。
解决
# 采用 先序遍历 递归形式。
def hasPathSum(self, root: TreeNode, targetSum: int) -> bool:
def preOrder(root: TreeNode, current_sum: int):
if root is not None:
# 获取 当前节点 的值。
current_sum += root.val
# 当前 叶子节点。
if root.left is None and root.right is None and current_sum == targetSum:
return True
return preOrder(root.left, current_sum) or preOrder(root.right, current_sum)
else:
# 空树 返回false。
return False
return preOrder(root, 0)
返回值处理:
return preOrder(root.left, current_sum) or preOrder(root.right, current_sum)