path sum

Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.

For example:
Given the below binary tree and  sum = 22 ,
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

return true, as there exist a root-to-leaf path 5->4->11->2 which sum is 22.

思路:典型的二叉树数据结构方法。非递归方式,可用两个栈,S和total,S记录节点,total记录从根节点到当前节点的和,当到达叶子节点时,判断其和是否等于sum。

class Solution {
public:
    bool hasPathSum(TreeNode *root, int sum) {
        stack<TreeNode*> S;
        stack<int> total;
        int val = 0;
        while(root != NULL || !S.empty())
        {
            while(root != NULL)
            {
                S.push(root);
                total.push(val + root->val);
                val = total.top();
                if (root->left == NULL && root->right == NULL) //是否到达叶子节点
                {
                    if (total.top() == sum)
                    {
                        return true;
                    }    
                }    
                root = root->left;
            }       
            root = S.top();
            S.pop();
            val = total.top();
            total.pop();
            root = root->right;
        }   
        return false;  
    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值