1. 程式人生 > 實用技巧 >129. 求根到葉子節點數字之和

129. 求根到葉子節點數字之和

給定一個二叉樹,它的每個結點都存放一個0-9的數字,每條從根到葉子節點的路徑都代表一個數字。
例如,從根到葉子節點路徑 1->2->3 代表數字 123。
計算從根到葉子節點生成的所有數字之和。
說明:葉子節點是指沒有子節點的節點。

輸入: [1,2,3]
    1
   / \
  2   3
輸出: 25
解釋:
從根到葉子節點路徑 1->2 代表數字 12.
從根到葉子節點路徑 1->3 代表數字 13.
因此,數字總和 = 12 + 13 = 25.
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
//dfs遞迴
class Solution {
    public int sumNumbers(TreeNode root) {
        return dfs(root,0);
    }

    public int dfs(TreeNode node,int presum){
        if(node==null)
            return 0;       
        int sum = node.val+presum*10; // 當前節點值+前一結點值*10
        if(node.left==null && node.right==null)//葉子節點返回sum
            return sum;
        return dfs(node.left,sum)+dfs(node.right,sum); //非葉子節點返回左右和
    }
}