算法 - 二叉树根节点到叶子节点的所有路径和

题目描述

给定一个仅包含数字 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；

---

最后，我要进大厂！加油奥里给！