Sum Root to Leaf Numbers

最新推荐文章于 2019-08-12 13:35:43 发布

原创最新推荐文章于 2019-08-12 13:35:43 发布 · 614 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Sum Root to Leaf Num #leetcode #DFS #深度优先

LeetCode 同时被 2 个专栏收录

44 篇文章

订阅专栏

C++Technology

7 篇文章

订阅专栏

本文介绍了一道LeetCode题目——求解二叉树从根节点到叶节点的所有路径所代表的数字之和，并提供了一个使用深度优先搜索实现的C++解决方案。

LeetCode 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.

深度优先搜索：

//将动态数组中的单个数数转换成整数
int vecToInt(vector<int>& v)
{
	int sum = 0;
	for(int i=0; i<v.size(); i++)
	{
		sum += (pow(10, v.size()-1-i)*v[i]);
	}

	return sum;
}

void pathSum(TreeNode* p, vector<int>& sub_sum, int & sum)
{
	//如果是叶节点，则计算出动态数组中的数字，计入sum中
	if(p->left==NULL && p->right==NULL)
	{
		sub_sum.push_back(p->val);
		sum = sum + vecToInt(sub_sum);
		return ;
	}
	
	sub_sum.push_back(p->val);
	if(p->left!=NULL)
	{
		pathSum(p->left, sub_sum, sum);
		sub_sum.pop_back();
	}

	if(p->right!=NULL)
	{
		pathSum(p->right, sub_sum, sum);
		sub_sum.pop_back();
	}
}

int sumNumbers(TreeNode *root) 
{
	if(root==NULL)
		return 0;

	vector<int> sub_sum;
	int sum = 0;

	pathSum(root, sub_sum, sum);
	return sum;
}