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

leetcode_129_Sum Root to Leaf Numbers

时间:2015-03-07 14:15:09      阅读:118      评论:0      收藏:0      [点我收藏+]

标签:leetcode   c++   tree   dfs   

欢迎大家阅读参考,如有错误或疑问请留言纠正,谢谢技术分享


Sum Root to Leaf Numbers

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.


class Solution {
public:
    int sumNumbers(TreeNode *root) {
		if(root!=NULL)
			sumNumbersUtil( root , 0 );
		return ans;
    }
private:
	int ans=0;
	void sumNumbersUtil(TreeNode *root , int cur)
	{
		if(!root)
			return;
		if(root->left!=NULL || root->right!=NULL)
		{
			sumNumbersUtil(root->left,  cur*10 + root->val);
			sumNumbersUtil(root->right, cur*10 + root->val);
		}
		else
			ans += cur*10 + root->val;
	}
};


//另一种写法
class Solution {
public:
    int sumNumbers(TreeNode *root) {
		int ans=0;
		if(root!=NULL)
			sumNumbersUtil( root , 0 , ans);
		return ans;
    }
private:
	
	void sumNumbersUtil(TreeNode *root , int cur ,int& ans)
	{
		if(!root)
			return;
		if(root->left!=NULL || root->right!=NULL)
		{
			sumNumbersUtil(root->left,  cur*10 + root->val , ans);
			sumNumbersUtil(root->right, cur*10 + root->val , ans);
		}
		else
			ans += cur*10 + root->val;
	}
};


leetcode_129_Sum Root to Leaf Numbers

标签:leetcode   c++   tree   dfs   

原文地址:http://blog.csdn.net/keyyuanxin/article/details/44115283

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