剑指offer(C++)-JZ82:二叉树中和为某一值的路径(一)(数据结构-树)

作者:翟天保Steven
版权声明:著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处

题目描述:

给定一个二叉树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

解题思路:

本题考察数据结构树的使用,可运用stack栈后入先出的特性解题。

1)将根节点放入栈进入循环。

2)获取根节点的值,并弹出根节点,当左子树存在时,将左子树的数值叠加根节点的数值,并将左子树放入栈,右子树同理。

3)从栈顶取出节点,该节点应该是上一轮的右子树,以其为该轮的根,继续扫描其左右子树,若有则继续存入栈内。

4)当栈为空时,说明所有的结点都遍历过了,无结果则返回false。

5)当某一轮根结点的值与目标值一致且左右子树均无,则说明该节点是叶子节点,且符合要求,返回true完成。

该逻辑类似于前序遍历,无非是将过程中每个节点的值替换为对应路径值的和。

测试代码:

/**
 * 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) {
        stack<TreeNode*> t;
        if(!root)
            return false;
        t.push(root);
        TreeNode* current;
        while(!t.empty())
        {
            current=t.top();
            t.pop();
            if(current->left)
            {
                current->left->val+=current->val;
                t.push(current->left);
            }
            if(current->right)
            {
                current->right->val+=current->val;
                t.push(current->right);
            }
            if(current->val==sum&&!current->left&&!current->right)
                return true;
        }
        return false;
    }
};

猜你喜欢

转载自blog.csdn.net/zhaitianbao/article/details/123145796