题目描述
给定一个仅包含数字 0-9 的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1 → 2 → 3 ,那么这条路径就用 123 来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如:
这颗二叉树一共有两条路径,
根节点到叶子节点的路径 1 → 2 用数字 12 代替
根节点到叶子节点的路径 1 → 3 用数字 13 代替
所以答案为 12+13=25
示例1 输入 {1,0} 返回值 10
示例2 输入 {1,#,9} 返回值 19
尝试一:
很显然这是一道关于 DFS(深度优先算法)的题目;肯定很简单啊!递归不就可以了!唯一不同的是对数字的处理,即从根节点到叶子节点的数据是 通过添加的方式,不是求和的方式!于是有一下代码:
public String sum1 = "";
public ArrayList<String> list1 = new ArrayList<>();
public int sumNumbers1(TreeNode root) {
// write code here
if (null == root) return 0;
if (null == root.left && null == root.right) {
list1.add(sum1 + root.val);
return Integer.parseInt(list1.get(list1.size() - 1));
}
sum1 += root.val;
if (root.left != null) sumNumbers1(root.left);
if (root.right != null) sumNumbers1(root.right);
sum1 = sum1.substring(0, sum1.length() - 1);
int res = 0;
Iterator<String> iterator = list1.iterator();
System.out.println(list1);
while (iterator.hasNext()) {
res += Integer.parseInt(iterator.next());
}
return res;
}
工具类就不必了,不过是个二叉树而已;运行可行,不过,时间复杂度较高!实际上,一开始 sum 我使用的不是 String,而是 StringBuilder ,不过改成 String 也并没有什么改观!
分析:一是使用 String 对数据处理的效率较低;二是因为最后又一步遍历的操作!
虽然使用的是递归,但是没有很好的理解递归的思想!你看,这里还是借助了集合
尝试二:
看了其他小伙伴的解答,优秀的地方在于对数据的处理不是使用 += 也不是使用 StringBuilder 的 append,而是使用 sum * 10 + val,这就很好的解决了这个问题!
于是有以下 “邯郸学步”代码:
int sum = 0;
int res = 0;
public int sumNumbers(TreeNode root) {
// write code here
if (null == root) return 0;
if (null == root.left && null == root.right) {
return sum * 10 + root.val;
}
sum = sum * 10 + root.val;
if (root.left != null) res += sumNumbers(root.left);
if (root.right != null) res += sumNumbers(root.right);
sum /= 10;
return res;
}
额……错误原因,递归没有整明白,哪里有这么 += ?这里重复 += 导致结果大于实际值
正文来了
看了看 解法,
public int sumNumbers(TreeNode root) {
if (root == null) {
return 0;
}
return sumNumbers(root, root.val);
}
/**
* 高手的递归!注意传入的参数!这样就解决了重复 += 的BUG
*/
private int sumNumbers(TreeNode root, int sum) {
if (root.left == null && root.right == null) {
return sum;
} else {
int result = 0;
if (root.left != null) {
result += sumNumbers(root.left, sum * 10 + root.left.val);
}
if (root.right != null) {
result += sumNumbers(root.right, sum * 10 + root.right.val);
}
return result;
}
}
传参是这个方法的灵魂!直不楞登的使用递归,硬递归还是不太好,要灵活多变,通过参数的传递,这样就解决了 重复 += 的 BUG;
最后,我要进大厂!加油奥里给!