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
.
思路:记录每条路径上的元素值,并将其转化成一个n位数。
添加一个辅助的向量v,记录到达某个结点p前,此路径上的所有元素值。当到达叶子结点时,将v中的元素都取出来,转化成一个n位数。
程序中的 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;
};