码迷,mamicode.com
首页 > 其他好文 > 详细

LC112 路径总和

时间:2020-07-16 21:33:07      阅读:50      评论:0      收藏:0      [点我收藏+]

标签:地方   入队   des   public   默认   return   pat   因此   dep   

终于自己做出来一道。。。

递归思路1

参考递归求解最大深度,构造新函数,将节点当前路径和当作额外参数传入
这个题比较特殊的地方在于,必须是叶子节点所在的路径才有效,因此在return true的条件中加入了left right均为`nullptr
返回时使用||不影响某个分支上的正确结果

class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        return depth(root, 0, sum);
    }
    bool depth(TreeNode* root, int res, int sum){
        if(root == nullptr)
            return false;
        //这两行注释也可以通过,不知道为什么,其实少了返回条件,猜测是因为函数没有返回值,而bool默认值为false
        //if(res + root->val == sum && root->left == nullptr && root->right == nullptr)
            //return true;  
        if(root->left == nullptr && root->right == nullptr)
            return (res + root->val) == sum;  
        bool lres = depth(root->left, res + root->val, sum);
        bool rres = depth(root->right, res + root->val, sum);
        return lres || rres;
    }
};

递归思路2

官方题解中采用的方法很巧妙,通过剩余值来代替上述方法中的路径和,直接在函数中递归即可

class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        if(root == nullptr)
            return false;
        if(root->left == nullptr && root->right == nullptr)
            return sum == root->val;
        return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
    }
};

非递归思路 广度优先搜索

使用两个队列,分别存储节点以及当前节点路径和,节点的子节点入队列时,当前节点的路径和与子节点的值和相加入队列

class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        if(root == nullptr)
            return false;
        queue<TreeNode*> nodes;
        queue<int> sums;
        nodes.push(root);
        sums.push(root->val);
        while(!nodes.empty()){
            TreeNode* nowNode = nodes.front();
            int nowSum = sums.front();
            if(nowNode->left == nullptr && nowNode->right == nullptr && nowSum == sum)
                return true;
            if(nowNode->left != nullptr){
                nodes.push(nowNode->left);
                sums.push(nowSum + nowNode->left->val);
            }
            if(nowNode->right != nullptr){
                nodes.push(nowNode->right);
                sums.push(nowSum + nowNode->right->val);
            }
            nodes.pop();
            sums.pop();
        }
        return false;
    }
};

LC112 路径总和

标签:地方   入队   des   public   默认   return   pat   因此   dep   

原文地址:https://www.cnblogs.com/imagineincredible/p/13323933.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!