Given a binary tree containing digits from0-9
only, each root-to-leaf path could represent a number.
An example is the root-to-leaf path1->2->3
which represents the number123
.
Find the total sum of all root-to-leaf numbers.
For example,
1
/ \
2 3
The root-to-leaf path1->2
represents the number12
.
The root-to-leaf path1->3
represents the number13
.
Return the sum = 12 + 13 =25
.
Solution 1: top down,往下传值,到leaf node时加到sum
int sum = 0;
public int sumNumbers(TreeNode root) {
if (root == null) {
return sum;
}
helper(root, 0);
return sum;
}
private void helper(TreeNode root, int num) {
num = num * 10 + root.val;
if (root.left == null && root.right == null) {
sum += num;
}
if (root.left != null) {
helper(root.left, num);
}
if (root.right != null) {
helper(root.right, num);
}
}
Solution 2: bottom up
public int sumNumbers(TreeNode root) {
return sum(root, 0);
}
public int sum(TreeNode n, int s){
if (n == null) return 0;
if (n.right == null && n.left == null) return s*10 + n.val;
return sum(n.left, s*10 + n.val) + sum(n.right, s*10 + n.val);
}