二叉樹根到葉子結點數字之和

題目描述：

認爲二叉樹的根到每個葉子結點的路徑上的點是一個數，若該條路徑有三個結點，則表示的數爲根結點節點的值乘100加下一結點值乘10加葉子結點的值。該問題是想求所有路徑構成數的和。

如上二叉樹，其三條路徑表示的數分別爲：124,135,136則應返回395。 

解決思路：

算法一：看到這種問題最初的想法，不考慮性能只考慮實現，上來二話不說、直接DFS，將所有路徑都求出來存入list<list>中，然後再遍歷路徑列表得到結果。

（ps：別問這塊爲啥沒代碼......）

算法二：將所有路徑都求出來這種做法真的很浪費空間，能否再此基礎上優化一下？使用DFS + 回溯，只需要存儲一條路經即可。

     public void selution(TreeNode root, List<Integer> path,int[] sum) {
         path.add(root.val);
         if(root.left == null && root.right == null) {
             sum[0] += getSum(path);
             path.remove(path.size() - 1);//恢復現場
             return;
         }
         if(root.left != null) selution(root.left, path, sum);
         if(root.right != null) selution(root.right, path, sum);
         path.remove(path.size() - 1);//恢復現場
     }
     public int getSum(List<Integer> path) {
         int sum = 0;
         for(int num : path) {
             sum = sum * 10 + num;
         }
         return sum;
     }

算法三：一條路徑都不想存，想要在常數空間複雜度條件下完成。在遞歸的過程輸人該該條路徑之前結點的值temp，先更新temp值爲temp * 10 + root.val。再判斷當前結點是否爲葉子結點，若是葉子結點則將temp加到和上即可，若非也葉子結點，則依照此法處理其左、右孩子即可。

該方法雖然沒有現場恢復操作，但是還是屬於回溯的，這是由於傳入的temp是int型，基本數據類型，傳入函數裏面的變化並不會影響外面的temp值。

具體實現代碼如下：

     public void selution(TreeNode root, int temp, int[] sum) {
         if(root == null){
             return;
         }
         temp = temp * 10 + root.val;
         if(root.left == null && root.right == null) {
             sum[0] += temp;
             return;
         }
         selution(root.left, temp, sum);
         selution(root.right, temp, sum);
         
     }
     public int sumNumbers(TreeNode root) {
         if(root == null){
             return 0;
         }
         int[] sum = new int[1];
         selution(root,0,sum);
         return sum[0];
     }

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。