LeetCode112. 路径总和

快乐早睡

于 2024-05-02 22:56:35 发布

阅读量881

点赞数 20

分类专栏： LeetCode 文章标签： C++ Java 数据结构算法二叉树递归深度优先搜索

本文链接：https://blog.csdn.net/weixin_40909461/article/details/138401845

版权

LeetCode 专栏收录该内容

132 篇文章 0 订阅

订阅专栏

在这里插入图片描述

LeetCode112. 路径总和

题目：

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

示例：

在这里插入图片描述

输入： $roo t = [5, 4, 8, 11, n u ll, 13, 4, 7, 2, n u ll, n u ll, n u ll, 1], t a r g e tS u m = 22$
输出： $t r u e$

解题思路一：(递归)

确定递归函数的参数和返回值：

bool dfs(TreeNode* cur, int cnt);

确定终止条件：

if(!cur->left && !cur->right && cnt == 0) return true;
if(!cur->left && !cur->right) return false;

确定单层递归逻辑： 先序遍历顺序为： $左 - > 中 - > 右$

if(cur->left){
    cnt -= cur->left->val;
    if(dfs(cur->left, cnt)) return true;
    cnt += cur->left->val;
}
if(cur->right){
    cnt -= cur->right->val;
    if(dfs(cur->right, cnt)) return true;
    cnt += cur->right->val;
}
return false;

C++版整体代码

class Solution {
public:
    bool dfs(TreeNode* cur, int cnt){
        if(!cur->left && !cur->right && cnt == 0) return true;
        if(!cur->left && !cur->right) return false;
        if(cur->left){
            cnt -= cur->left->val;
            if(dfs(cur->left, cnt)) return true;
            cnt += cur->left->val;
        }
        if(cur->right){
            cnt -= cur->right->val;
            if(dfs(cur->right, cnt)) return true;
            cnt += cur->right->val;
        }
        return false;
    }
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root == NULL) return false;
        return dfs(root, targetSum - root->val);
    }
};

Java版整体代码

class Solution {
    public boolean dfs(TreeNode cur, int cnt){
        if(cur.left == null && cur.right == null && cnt == 0) return true;
        if(cur.left == null && cur.right == null) return false;
        if(cur.left != null){
            cnt -= cur.left.val;
            if(dfs(cur.left, cnt) == true) return true;
            cnt += cur.left.val; 
        }
        if(cur.right != null){
            cnt -= cur.right.val;
            if(dfs(cur.right, cnt) == true) return true;
            cnt += cur.right.val;
        }
        return false;
    }
    public boolean hasPathSum(TreeNode root, int targetSum) {
        if(root == null) return false;
        return dfs(root, targetSum - root.val);
    }
}