LeetCode sum-root-to-leaf-numbers

题目描述

给定一个仅包含数字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);
    }
};