112. 路径总和

leecode原题

题目

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径，这条路径上所有节点值相加等于目标和 targetSum 。如果存在，返回 true ；否则，返回 false 。

叶子节点: 是指没有子节点的节点。

示例

示例 1:

输入：root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出：true
解释：等于目标和的根节点到叶节点路径如上图所示。

示例 2:

输入：root = [1,2,3], targetSum = 5
输出：false
解释：树中存在两条根节点到叶子节点的路径：
(1 --> 2): 和为 3
(1 --> 3): 和为 4
不存在 sum = 5 的根节点到叶子节点的路径。

示例 3:

输入：root = [], targetSum = 0
输出：false
解释：由于树是空的，所以不存在根节点到叶子节点的路径。

提示：

• 树中节点的数目在范围 [0, 5000] 内
• -1000 <= Node.val <= 1000
• -1000 <= targetSum <= 1000

解题思路

思路

这题跟257. 二叉树的所有路径类似，也是可以采用前序遍历实现, 只需要到叶子节点出判断累加和是否等于目标值。

实现

源码

func travel(root *TreeNode, targetSum int, sum int) bool {
    if root == nil {
        return false
    }
    // 是叶子节点
    if root.Left == nil && root.Right == nil {
        sum += root.Val
        if sum == targetSum {
            return true
        }
        return false
    }
    sum += root.Val
    return travel(root.Left, targetSum, sum) || travel(root.Right, targetSum, sum)
}

func hasPathSum(root *TreeNode, targetSum int) bool {
    sum := 0
    return travel(root, targetSum, sum)
}