LeetCode129. 求根到葉子節點數字之和
阿新 • • 發佈:2018-12-11
給定一個二叉樹,它的每個結點都存放一個 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
.
思路:深度優先遍歷二叉樹。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public int sumNumbers(TreeNode root) { List<Integer> res=new LinkedList<Integer>();//儲存根結點到葉子結點的所有路徑值 String tmp="";//臨時儲存根結點到葉子結點的路徑值 dfs(root,res,tmp);//深度優先遍歷二叉樹 int sum=0;//所有路徑和 for(Integer i:res){ sum+=i; } return sum; } public void dfs(TreeNode root, List<Integer> res,String tmp){ if(root==null){ return ; } if(root.left==null&&root.right==null){ //葉子結點 tmp+=root.val; int n=Integer.parseInt(tmp); res.add(n); return ; } tmp+=root.val; dfs(root.left,res,tmp); tmp.substring(0,tmp.length()-1); dfs(root.right,res,tmp); tmp.substring(0,tmp.length()-1); } }