路径总和

85 阅读1分钟

题目描述

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

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

示例

image.png

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

思路

  1. 确定遍历顺序:前中后序均可,因为不涉及中间节点的处理
  2. 终止条件,遍历到叶子节点
  3. 思路:设置sum变量保存路径和,sum初始值为root的值,遍历到叶子节点时判断sum和targetSum是否相等。
  4. 注意回溯

代码

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function hasPathSum(root: TreeNode | null, targetSum: number): boolean {
    if(root){
        return getSum(root,root.val,targetSum)
    }
    return false

};
// 采用前序遍历,参数:node,sum,返回值:boolean
// 终止条件:遍历到叶子节点
// 单层逻辑:
// 先遍历左边,再遍历右边,注意回溯

function getSum(node: TreeNode | null, sum: number,targetSum): boolean{
    // 终止条件
    if(node!=null&&node.left==null&&node.right==null){
        if(sum===targetSum){
            return true;
        }else{
            return false;
        }
    }

    // 单层递归逻辑
    if(node&&node.left){
        sum+=node.left.val;
        if(getSum(node.left,sum,targetSum)) return true;
        sum-=node.left.val;
    }

    if(node&&node.right){
       sum+=node.right.val;
        if(getSum(node.right,sum,targetSum)) return true;
        sum-=node.right.val;
    }
    return false;
}

image.png

总结

如有错误之处,欢饮大家留言指教,谢谢大家了!