力扣【二叉树专题】👊 112. 路径总和

192 阅读2分钟

「这是我参与2022首次更文挑战的第 16 天,活动详情查看:2022首次更文挑战

题目链接

112. 路径总和 - 力扣(LeetCode) (leetcode-cn.com)

题目描述

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

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

测试用例

示例 1:

image.png

输入: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

题目分析

题目会提供一个二叉树和目标值,需要我们检查是否存在这么一条路径,使得从根到一个叶子节点并且包括途径的全部点的值之和为目标值

换个说法,需要我们对树进行深度搜索,从根节点起,累加每个节点的值,当搜索到叶子节点的时候,需要判断当前和的累加值是否就是目标值;因为是对树进行深度搜索,且每次判断目标值的操作在叶子节点,这里就存在一个细节,当我们在做回溯操作的时候,就需要从累加值中减去对应节点的值

image.png

使用示例的二叉树,为其标记上遍历的顺序,橘色表示深入遍历,红色表示回溯的轨迹,这种遍历的方式其实就是对二叉树的中序遍历,用 js 风格的代码来描述如下:

trave(root);
function trave(node) {
    if (node == null) return;
    // 遍历到一个节点时需要做些什么事情
    trave(node.left);
    // 当这个节点的左子树遍历完成之后,又需要做些什么
    trave(node.right);
    // 当这个节点的右子树遍历完成之后,又需要做什么
    // 在这一步之后,就是回溯到上一个节点
}

把我们的分析代入到上面的模板,代码就出来了

代码实现

var hasPathSum = function(root, targetSum) {
    let find = false;
    let sum = 0;
    trave(root);
    return find;

    function trave(node) {
        if (node == null || find) return;
        sum += node.val;
        trave(node.left);
        if (sum == targetSum && node.left == null && node.right == null) {
            find = true;
        }
        trave(node.right);
        sum -= node.val;
    }
};

image.png

当我们捋清楚二叉树的结构以及递归的回溯,就能很轻易的解决这题