Leetcode 129. 求根到叶子节点数字之和

题目

给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。

例如，从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答

其实就是个 前序遍历题 而已。

要点如下：

1. 在到达叶子结点时，才将此路径积累成的数字返回。
2. 到达非叶子节点时，加上当前非叶子结点的值之后递归左右子树。

有兴趣的同学可以改成非递归版的。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    int res = 0;
    public int sumNumbers(TreeNode root) {
        sumNumbers(root, 0);
        return res;
    }
    
    private void sumNumbers(TreeNode node, int num) {
        if(node == null) return;
        
        num = num * 10 + node.val;
        if(node.left == null && node.right == null) {
            res += num;
        }
        
        sumNumbers(node.left, num);
        sumNumbers(node.right, num);
    }
}

另一种写法

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    public int sumNumbers(TreeNode root) {
        return sumNumbers(root, 0);
    }
    
    private int sumNumbers(TreeNode node, int num) {
        if(node == null) return 0;
        
        num = num * 10 + node.val;
        if(node.left == null && node.right == null) {
            return num;
        }
        
        return sumNumbers(node.left, num) + sumNumbers(node.right, num);
    }
}

结果