求根到叶子节点数字之和

给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。

例如,从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

示例 1:

输入: [1,2,3]
    1
   / \
 2   3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此,数字总和 = 12 + 13 = 25.
示例 2:

输入: [4,9,0,5,1]
   4
   / \
  9   0
 / \
5   1
输出: 1026
解释:
从根到叶子节点路径 4->9->5 代表数字 495.
从根到叶子节点路径 4->9->1 代表数字 491.
从根到叶子节点路径 4->0 代表数字 40.
因此,数字总和 = 495 + 491 + 40 = 1026.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

 

利用后序遍历的非递归实现改造得到

后序遍历的非递归是利用栈来实现的,每一个栈顶元素之下的元素都是当前节点到跟节点的逆路径,然后利用这个性质通过判断 当前的节点是不是叶子节点,然后根据当前的栈得到具体的数值

 

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int sumNumbers(TreeNode* root) 
    {
        if(root == NULL)
        {
            return 0;
        }
        TreeNode* p = root;
        TreeNode* r;
        int sum = 0;
        bool flag;
        stack<TreeNode*> s1;
        stack<TreeNode*> s2;
        do
        {
            while(p != NULL)
            {
                s1.push(p);
                s2.push(p);
                p = p->left;
            }  //找到最左下角的节点,这个节点没有左子树
           
            r = NULL;
            flag = true;
            //cout<<"s1.size() = "<<s1.size()<<endl;
            //cout<<"s2.size() = "<<s2.size()<<endl;
            
            while(!s1.empty() && flag)
            {
                p = s1.top();
                if(p->right == r)
                {
                    if(p->left == NULL && p->right == NULL)     //叶子节点
                    {
                        //string tmp;
                        int numtmp = 0;
                        for(int i = 0; i<s1.size(); i++)
                        {
                            //tmp = tmp + s2.pop->val;
                            TreeNode* tmp = s2.top();
                            
                            numtmp = numtmp + (tmp->val)*pow(10,i);
                            //cout<<"numtmp = "<<numtmp<<endl; 
                            s2.pop();
                        }
                        
                        sum = sum + numtmp;
                        //cout<<"sum = "<<sum<<endl;
                        
                    }
                    s2 =s1;
                    p = s1.top();
                    r = p;
                    s1.pop();
                    s2.pop();
                    
                    
                }
                else
                {
                     p = p->right;
                     flag = false;
                }
                
            }
            // return 1;
            
        }while(!s1.empty());
        
        return sum;
    }
};

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值