【LeetCode】129. 求根到叶子节点数字之和（Java）

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

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

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

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

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int sum = 0;    // 总和，返回值

    public int sumNumbers(TreeNode root) {
    	// dfs
        // 先判断root是否为空，防止进入dfs方法后空指针异常
        if (root == null)   return 0;

        dfs(root, 0);
        return sum;
    }

    private void dfs(TreeNode tree, int curSum) {
        // 前面的总和乘10后加上当前值
        curSum = curSum * 10 + tree.val;
        // 如果左子树不为空，递归
        if (tree.left != null)  dfs(tree.left, curSum);
        // 如果右子树不为空，递归
        if (tree.right != null)  dfs(tree.right, curSum);
        // 如果左右子树都为空，说明到达叶子节点，总和sum加上当前总和
        if (tree.left == null && tree.right == null)  sum += curSum;
    }
}