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

本文介绍了在二叉树中查找和为特定值的路径的两种方法:递归和非递归(使用栈和深度优先搜索)。递归通过逐步减去节点值寻找路径,而非递归则通过增加节点值并存储路径和进行查找。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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

题目链接:二叉树中和为某一值的路径(一)

解题思路1:递归

我们或许想记录下每一条从根节点到叶子节点的路径,计算出该条路径的和,但此种思路用递归稍麻烦,我们可以试着把和转换为差,从根节点开始,sum就减去根节点的值,如果到叶子节点,减为0了,那就证明此路径满足要求

递归结束条件:节点为空,意味着已经走过了叶子节点,返回;每当检查到某节点没有子节点,则说明该节点为叶子节点,如果此时sum的值减去该节点的值刚好为0,则说明找到了路径

返回值:将子问题中是否有复合新目标值的路径层层往上返回

本级任务:每一层需要检查是否到了叶子节点,如果没有则递归进入子节点,同时更新sum值减掉本层的节点值

代码如下:

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

解题思路2:非递归,栈+DFS

我们上面是一步一步减去当前root的值,看最后root为空时,sum的值是否为0,采用递归实现,在此,我们采用走一步加上一步当前节点的值,看最后加和是否满足sum,我们采用栈和深度优先搜索来实现

深度优先搜索:深度优先搜索一般用于树或者图的遍历,其他有分支也适用,它是从初始点开始,一直沿着同一个分支遍历,直到该分支结束,然后回溯到上一级继续沿着一个分支走到底,如此往复,直到所有的节点都有被访问到

具体步骤:

首先检查空节点,若为空树,则返回

使用栈嵌套pair,first记录节点,second记录经过的节点的值的和。根节点及根节点的值先进栈。

遍历的时候判断是否是叶子节点,且和是否等于sum,是则返回true

如果节点有子节点,就入栈,并更新路径和

如果遍历结束也没有找到规定的路径,则返回false,该二叉树中没有符合条件的路径

当到达当前节点时,second中的值已经加过该节点的值了

代码如下:

    bool hasPathSum(TreeNode* root, int sum) {
        if (root == nullptr) return false;
        //栈辅助深度优先遍历,并记录到相应节点的路径和
        stack<pair<TreeNode*, int>> s;
        //根节点先入栈
        s.push({root, root->val});
        while (!s.empty()) {
            auto temp = s.top();
            s.pop();
            //是叶子节点,且路径和等于sum
            if (temp.first->left == nullptr && temp.first->right == nullptr &&
                    temp.second == sum)
                return true;
            //左节点入栈
            if (temp.first->left != nullptr)
                s.push({temp.first->left, temp.first->left->val + temp.second});
            //右节点入栈
            if (temp.first->right != nullptr)
                s.push({temp.first->right, temp.first->right->val + temp.second});
        }
        return false;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值