LeetCode 112. 路径总和（深度优先搜索dfs、二叉树）

岁忧

已于 2024-08-08 00:01:28 修改

阅读量224

点赞数 5

分类专栏： LeetCode 文章标签：深度优先 leetcode 算法 c++ 数据结构

于 2024-08-07 23:56:54 首次发布

本文链接：https://blog.csdn.net/weixin_46028214/article/details/141004067

版权

LeetCode 专栏收录该内容

544 篇文章 1 订阅

订阅专栏

112. 路径总和

在这里插入图片描述

思路：需要注意，要在叶子节点的时候进行判断，判断的时候不要忘记加上叶子节点的值。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool flag=0;
    void dfs(TreeNode* root,int sum, int targetSum){
    	//在叶子节点的时候进行判断
        if(root->left==nullptr && root->right==nullptr){
            if(sum==INT_MIN) sum=0;
            //判断的时候不要忘记加上叶子节点的值。
            if(sum+root->val==targetSum) flag=1;
            return ;
        }
        if(sum==INT_MIN) sum=0;
        if(root->left)  dfs(root->left,sum+root->val,targetSum);
        if(root->right) dfs(root->right,sum+root->val,targetSum);
        
    }
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root) dfs(root,INT_MIN,targetSum);
        return flag;
    }
};

优化了一下，从targetSum开始减

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool flag=0;
    void dfs(TreeNode* root, int targetSum){
        if(root->left==nullptr && root->right==nullptr){
            if(root->val==targetSum) flag=1;
            return ;
        }
        if(root->left)  dfs(root->left,targetSum-root->val);
        if(root->right) dfs(root->right,targetSum-root->val);
        
    }
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root) dfs(root,targetSum);
        return flag;
    }
};