题目描述
给定一个仅包含数字0-9的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1->2->3,那么这条路径就用123来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如:
1↵ / ↵ 2 3
根节点到叶子节点的路径1->2用数字12代替
根节点到叶子节点的路径1->3用数字13代替
所以答案为12+13=25
解题思路
采用递归方法,每个非叶子节点都有两条方向,结果就是两个方向的sum的和。
如果只有单个子树,则缺失的那个子树不存在路径。
当碰到叶子结点就返回sum。
/**
* 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 pre(TreeNode *root, int sum){
if(root == NULL)
return 0;
sum = sum * 10 + root->val;
if(root->left == NULL && root->right == NULL) //叶子节点
return sum;
return pre(root->left, sum) + pre(root->right, sum);
}
int sumNumbers(TreeNode *root) {
return pre(root, 0);
}
};