【剑指Offer】JZ82 二叉树中和为某一值的路径(一)

20 篇文章 0 订阅

题目地址:牛客网

题目:

描述

给定一个二叉树root和一个值 sum ,判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。

1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点

2.叶子节点是指没有子节点的节点

3.路径只能从父节点到子节点,不能从子节点到父节点

4.总节点数目为n


例如:
给出如下的二叉树,sum=22,


返回true,因为存在一条路径 5→4→11→2的节点值之和为 22

数据范围:

1.树上的节点数满足 0≤n≤10000

2.每 个节点的值都满足 |val|≤1000

要求:空间复杂度 O(n),时间复杂度 O(n)

进阶:空间复杂度 O(树的高度),时间复杂度 O(n)

示例1

输入:

{5,4,8,1,11,#,9,#,#,2,7},22

返回值:

true

示例2

输入:

{1,2},0

返回值:

false

示例3

输入:

{1,2},3

返回值:

true

示例4

输入:

{},0

返回值:

false

 解题思路:

注意一下题目的描述,只能从根结点到叶子结点,而不是到任何结点。

这样的话其实就比较简单了,不用反复回溯。只需要一边层次遍历就行。每到一个结点就让它加上它父亲结点的值,直到遍历到叶子结点,判断它是否大于sum值。

我的代码部分其实还可以优化,比如当一个结点大于sum了就可以剪枝,但这题不剪枝也能过。

代码:

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return bool布尔型
     */
    bool hasPathSum(TreeNode* root, int sum) {
        // write code here
        TreeNode* p=root;
        if(!p)return false;
        queue<TreeNode*> q;
        q.push(p);
        while(q.size())
        {
            p=q.front();
            q.pop();
            if(!p->left&&!p->right){
                if(p->val==sum)return true;
            }
            if(p->left){
                p->left->val=p->val+p->left->val;
                q.push(p->left);
            }
            if(p->right){
                p->right->val=p->val+p->right->val;
                q.push(p->right);
            }
            
        }
        return false;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值