码迷,mamicode.com
首页 > 其他好文 > 详细

【leetcode】Sum Root to Leaf Numbers(hard)

时间:2015-04-30 19:47:36      阅读:107      评论:0      收藏:0      [点我收藏+]

标签:

Given a binary tree containing digits from 0-9 only, each root-to-leaf path could represent a number.

An example is the root-to-leaf path 1->2->3 which represents the number 123.

Find the total sum of all root-to-leaf numbers.

For example,

    1
   /   2   3

 

The root-to-leaf path 1->2 represents the number 12.
The root-to-leaf path 1->3 represents the number 13.

Return the sum = 12 + 13 = 25.

 

思路:最近睡得不好,搞得脑子也短路了。一直想着怎么从叶节点往上获取数字。费了好大的劲才AC了,结果一看答案,立马郁闷了。不到10行就能搞定。

具体的原理是深度优先,在向下迭代的过程中,用一个变量不断的把父节点以上的数字传下来,返回的时候把结果加起来。

public int sumNumbers(TreeNode root) {
        return helper(root, 0);
}

public int helper(TreeNode root, int curSum) {
        if(root == null) return 0;
        curSum = curSum*10 + root.val;
        if(root.left == null && root.right == null) return curSum;
        return helper(root.left, curSum) + helper(root.right, curSum);
}

 

 

下面是我自己AC的代码,超长,超繁琐,当个反面例子吧。思路是把每个数字都存在一个vector里面。之所以繁琐是因为我只是在返回的时候下功夫。

int sumNumbers(TreeNode *root) {
        int s, e, sum = 0;
        vector<vector<int>> v;
        Numbers(root, s, e, v);
        for(vector<vector<int>>::iterator it = v.begin(); it != v.end(); ++it)
        {
            int num = 0;
            while(!it->empty())
            {
                num = num * 10 + it->back();
                it->pop_back();
            }
            sum += num;
        }
        return sum;
    }

    void Numbers(TreeNode * root, int &s, int &e, vector<vector<int>>& v)
    {
        int sl = -1, el = -1, sr = -1, er = -1;
        s = e = -1;
        if(root == NULL) return;
        if(root->left == NULL && root->right == NULL) //在叶节点 压入新的数字
        {
            e = s = v.size();
            v.push_back(vector<int>(1,root->val));
            return;
        }
        if(root->left != NULL)
        {
            Numbers(root->left, sl, el, v);
            for(int i = sl; i <= el; ++i)
            {
                v[i].push_back(root->val);
            }
            s = sl; e = el;
        }
        if(root->right != NULL)
        {
            Numbers(root->right, sr, er, v);
            for(int i = sr; i <= er; ++i)
            {
                v[i].push_back(root->val);
            }
            s = (s == -1) ? sr : s;
            e = er;
        }
    }

 

【leetcode】Sum Root to Leaf Numbers(hard)

标签:

原文地址:http://www.cnblogs.com/dplearning/p/4469468.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!