题目描述
给定一个仅包含数字0-9的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1->2->3,那么这条路径就用123来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如:
1↵ / ↵ 2 3
根节点到叶子节点的路径1->2用数字12代替
根节点到叶子节点的路径1->3用数字13代替
所以答案为12+13=25
思路
又是一个深度优先搜索,找到叶子节点之后,就把当前的路径上代表的数加起来。
利用全局变量sum 和cur,不断维护这两个变量即可。注意要适时弹出cur中的元素。
代码
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int sum = 0;
vector<int> cur;
int sumNumbers(TreeNode *root) {
if (!root)
return 0;
else{
dfs(root);
return sum;
}
}
void dfs(TreeNode *root){
cur.push_back(root->val);
if (!root->left && !root->right){
for (int i=0;i<cur.size();i++){
sum += pow(10,(cur.size()-i-1))*cur[i];
}
return;
}
if (root->left){
dfs(root->left);
cur.pop_back();
}
if (root->right){
dfs(root->right);
cur.pop_back();
}
}
};