【Leetcode】129. Sum Root to Leaf Numbers

题目地址：

https://leetcode.com/problems/sum-root-to-leaf-numbers/

求所有从根到叶子节点拼成的数字的总和。

法1：DFS。如果是空树则直接返回。接着从root开始DFS，每次进入下一层递归的时候传一下从树根走到其父亲是什么数，比如这个数是$x$，那么走到当前节点$v$产生的数就是$10x+v$。如果走到叶子，就累加答案。代码如下：

public class Solution {
    
    private int res;
    
    public int sumNumbers(TreeNode root) {        
        dfs(root, 0);
        return res;
    }
    
    private void dfs(TreeNode root, int cur) {
        if (root == null) {
            return;
        }
        
        cur = cur * 10 + root.val;
        // 走到叶子了，累加答案
        if (root.left == null && root.right == null) {
            res += cur;
            return;
        }
        
        dfs(root.left, cur);
        dfs(root.right, cur);
    }
}

时间复杂度$O(n)$，空间$O(h)$。

法2：分治法。将树根到当前节点父亲拼成的数作为参数传递下去，直到叶子节点再返回。先求当前节点到左子树叶子组成的所有数之和，再求当前节点到右子树叶子组成的所有数之和，最后再加总。代码如下：

public class Solution {
    public int sumNumbers(TreeNode root) {
        return dfs(root, 0);
    }
    
    // 返回的是从原树树根走到cur为根的子树的所有叶子的数字总和
    private int dfs(TreeNode cur, int prev) {
        if (cur == null) {
            return 0;
        }
        
        int val = prev * 10 + cur.val;
        if (cur.left == null && cur.right == null) {
            return val;
        }
		
		// 累加左右答案返回        
        return dfs(cur.left, val) + dfs(cur.right, val);
    }
}

时空复杂度一样。