LeetCode | Sum Root to Leaf Numbers

最新推荐文章于 2023-10-10 09:42:58 发布

原创最新推荐文章于 2023-10-10 09:42:58 发布 · 563 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #二叉树

LeetCode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种算法，用于计算给定二叉树中从根节点到叶子节点的路径所代表的数字之和。通过记录路径上的每个节点值并转换为数字，最后累加得到总和。

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.

思路：记录每条路径上的元素值，并将其转化成一个ｎ位数。

添加一个辅助的向量ｖ，记录到达某个结点ｐ前，此路径上的所有元素值。当到达叶子结点时，将ｖ中的元素都取出来，转化成一个ｎ位数。

程序中的　void cal(vector<int> &v)；　最后将这个数累加到sum中即可。程序中的遍历采用递归的方法。

class Solution {
public:
	Solution():sum(0){}

    int sumNumbers(TreeNode *root) 
	{
		vector<int> v;
		if(!root)
			return sum;
		else
			path(root,v);
		return sum;
    }

	void path(TreeNode* root, vector<int> v)//遍历
	{
		v.push_back(root->val);
		if(!root->left && !root->right)//叶子结点
			cal(v);
		else if(root->left && !root->right)
			path(root->left,v);
		else if(root->right && !root->left)
			path(root->right,v);
		else
		{
			path(root->left,v);
			path(root->right,v);
		}
	}

	void cal(vector<int> &v)
	{
		int i;
		for(i=0; i<v.size(); ++i)//转化成n位数，并累加到sum中
			sum +=v[i]*Pow(10,v.size()-1-i);		
	}
	
	int Pow(int p, int n)//<span style="font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;">转化成n位数</span>
	{
		if(n==0)
			return 1;
		else if(n==1)
			return 10;
		return p*Pow(10,n-1);
	}

private:
	int sum;
};