题目:
给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。
例如,从根到叶子节点路径 1->2->3 代表数字 123。
计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此,数字总和 = 12 + 13 = 25.
来源:力扣(LeetCode)
解:
这道题显然就是深度遍历的考察。那么问题就在于怎么存储遍历值,以及怎么回溯?
比较直接的,就是弄一个String,然后+上去,最后一个Integer.intValue(string),转换成数字。但是这样浪费空间,毕竟你每加一个数字就得新建一个字符串,效率也低。
从根节点开始,作为第一位,那么每一次向下遍历,原值*10+this.val,就相当于求和了。为了回溯方便,直接用值传递即可,省去回传修改。
class Solution {
private int sum=0;
public void fc(TreeNode root , int val){
val*=10;
val+=root.val;
//叶子节点
if(root.left==null && root.right==null){
sum+=val;
return;
}
//非叶子,继续向下
if(root.left!=null){
fc(root.left,val);
}
if(root.right!=null){
fc(root.right,val);
}
}
public int sumNumbers(TreeNode root) {
if(root!=null){
fc(root,0);
}
return sum;
}
}