Leetcode——129. 求根到葉子節點數字之和(遞迴)
阿新 • • 發佈:2020-09-01
@目錄
129. 求根到葉子節點數字之和
題目**
給定一個二叉樹,它的每個結點都存放一個 0-9 的數字,每條從根到葉子節點的路徑都代表一個數字。 例如,從根到葉子節點路徑 1->2->3 代表數字 123。 計算從根到葉子節點生成的所有數字之和。 說明: 葉子節點是指沒有子節點的節點。 示例 1: 輸入: [1,2,3] 1 / \ 2 3 輸出: 25 解釋: 從根到葉子節點路徑 1->2 代表數字 12. 從根到葉子節點路徑 1->3 代表數字 13. 因此,數字總和 = 12 + 13 = 25. 示例 2: 輸入: [4,9,0,5,1] 4 / \ 9 0 / \ 5 1 輸出: 1026 解釋: 從根到葉子節點路徑 4->9->5 代表數字 495. 從根到葉子節點路徑 4->9->1 代表數字 491. 從根到葉子節點路徑 4->0 代表數字 40. 因此,數字總和 = 495 + 491 + 40 = 1026.
思想**
所有數字之和,可以用前序遍歷的遞迴模板,對於當前結點數字的處理:將之前的和pre_sum*10加上當前結點的值root.val即為當前總和cur_sum;如果該結點為葉子結點,則返回即可,否則遞迴呼叫訪問當前結點的子結點
程式碼**
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public int sumNum(TreeNode root, int pre_sum) { if(root == null){ return 0; } int cur_sum = pre_sum * 10 + root.val; if(root.left == null && root.right == null){ return cur_sum; } return sumNum(root.left, cur_sum) + sumNum(root.right, cur_sum); } public int sumNumbers(TreeNode root) { return sumNum(root, 0); } }