给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。计算从根节点到叶节点生成的所有数字之和 。
中等难度。使用前序遍历递归即可解决,比较简单。定义一个全局变量sum记录每一条路径的和。注意每次递归的路径和为10 * num + root.val。
int sum = 0;
public int sumNumbers(TreeNode root) {
dfs(root, 0);
return sum;
}
public void dfs(TreeNode root, int num) {
if (root == null) {
return;
}
//如果是叶子节点,那么不必再递归了
if (root.left == null && root.right == null) {
//sum等于之前的其他路径的sum和加上当前路径的sum
sum += num * 10 + root.val;
return;
}
//前序递归左右子节点,之前的值需要乘以10再加上当前节点的值
dfs(root.left, 10 * num + root.val);
dfs(root.right, 10 * num + root.val);
}