leetcode之sum-root-to-leaf-numbers

LeetCode之sum-root-to-leaf-numbers

题目

Given a binary tree containing digits from0-9only, each root-to-leaf path could represent a number.

An example is the root-to-leaf path1->2->3which represents the number123.

Find the total sum of all root-to-leaf numbers.

For example,

The root-to-leaf path1->2represents the number12.
The root-to-leaf path1->3represents the number13.

Return the sum = 12 + 13 =25.

题意

给你一个二叉树，二叉树节点都是数字组成，数字仅限于从0-9，每个根节点到叶子节点代表一个数。
例如：

这里的左边1-2 就是代表数字12；
这里右边的1-3就是代表数字13；
现在给定一个这样的二叉树，求这个二叉树最终代表的数字是多少？

解题思路

这里明显可以用二叉树的递归前序遍历完成。每次递归时，只需要将上一次的sum*10 再加上当前的本次计算的值就可以，以此循环往复，只要没有叶子节点，递归遍历完成之后，返回sum就好。

C++实现代码

/**
 * 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 sumNumbers(TreeNode *root) {
        int sum=0;
        if(root ==nullptr)
            return sum;
        return  preOrderSumNum(root,sum);
    }
    int preOrderSumNum(TreeNode *root,int sum){
        if(root ==nullptr)
            return 0;
        sum = sum*10+root->val;
        if(root->left == nullptr && root->right == nullptr)
            return sum;
        return preOrderSumNum(root->left,sum)+preOrderSumNum(root->right,sum);
    }
};