LeetCode112、 路径总和

LeetCode 系列记录我学习算法的过程。

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第16天，点击查看活动详情

题目

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

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

示例：

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

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

提示：

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

思路

所有能用递归来实现的题目，思路其实都差不多，只是要找到其中的规律：

• 第一步找结束条件，首先判断根节点是否存在，不存在直接返回 false
• 然后再判断左右子树均不存在的情况，返回 根节点值 与 target 是否相等
• 第二步考虑当前该做什么，默认当前递归函数已经实现了判断二叉树是否有路径总和与target相等，我们直接调用函数分别传入左右子树及处理后的目标值
• 最后返回左右子树递归的结果的 逻辑或 的值即可

代码实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} targetSum
 * @return {boolean}
 */
var hasPathSum = function(root, targetSum) {
    // 根节点不存在直接返回 false
    if (!root) return false
    
    // 左右子树不存在，返回 根节点的值 与 target 是否相等
    if (!root.left && !root.right) 
        return root.val === targetSum
        
    // 递归获取左右子树是否有路径与 目标值减去根节点值 后的结果相等
    let left = hasPathSum(root.left, targetSum - root.val)
    let right = hasPathSum(root.right, targetSum - root.val)
    
    // 返回结果
    return left || right
};