129. 求根到葉子節點數字之和
阿新 • • 發佈:2020-10-29
給定一個二叉樹,它的每個結點都存放一個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); //非葉子節點返回左右和 } }