【算法题】二叉树中和为某一值的路径(一)

二叉树中和为某一值的路径(一)

https://www.nowcoder.com/practice/508378c0823c423baa723ce448cbfd0c?tpId=196&tqId=37053&rp=1&sourceUrl=%2Fexam%2Foj%3FquestionJobId%3D10%26subTabName%3Donline_coding_page&difficulty=undefined&judgeStatus=undefined&tags=&title=

这题让判断的是从根节点到叶子节点的所有路径中,有没有和等于 sum 的路径,我们使用 DFS 从根节点开始遍历,每到一个节点就用 sum 减去该节点的值,如果到叶子节点, sum 正好等于叶子节点的值,则存在这样的路径。

如下图所示,有两个叶子节点的值等于 sum ,说明有两条这样的路径,实际上只要存在一条即可返回 true

alt

    public boolean hasPathSum(TreeNode root, int sum) {
        if (root == null)
            return false;
        // 如果到叶子节点了,并且剩余值等于叶子节点的值,返回true。
        if (root.left == null && root.right == null && sum == root.val)
            return true;
        // 分别沿着左右子节点走下去,然后顺便把当前节点的值减掉,
        // 左右子节点只要有一个返回true,说明存在这样的结果。
        return hasPathSum(root.left, sum - root.val)
                || hasPathSum(root.right, sum - root.val);
    }

public:
    bool hasPathSum(TreeNode *root, int sum) {
        if (root == nullptr)
            return false;
        // 如果到叶子节点了,并且剩余值等于叶子节点的值,返回true。
        if (root->left == nullptr && root->right == nullptr && sum == root->val)
            return true;
        // 分别沿着左右子节点走下去,然后顺便把当前节点的值减掉,
        // 左右子节点只要有一个返回true,说明存在这样的结果。
        return hasPathSum(root->left, sum - root->val)
               || hasPathSum(root->right, sum - root->val);
    }

各大厂算法面试题已经整理好了,请看这里:《算法专栏》

#笔试#

为了应对春招和秋招找工作,我经过长时间的收集和整理了各大厂的算法面试题,所有的算法题我都已经做了实现,大家可以根据自己需要面试的大厂选择练习即可。适宜人群: 在校生、社招求职者及自学者。

全部评论

相关推荐

03-16 22:00
武汉大学 C++
幸福的小熊猫想要offer:我阿里投的 c++岗,面试官说自己是做 java 的,c++这辈子才有了
点赞 评论 收藏
分享
评论
1
1
分享

创作者周榜

更多
牛客网
牛客企业服务