给定一个二叉树,它的每个结点都存放一个 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;
}
}