給一棵二叉樹,找出從根節點到葉子節點的所有路徑。
樣例
給出下面這棵二叉樹:
1
/ \
2 3
\
5
所有根到葉子的路徑為:
[
"1->2->5",
"1->3"
]
解:很經典的一道題,很簡單但是還是有一些注意的點。
先上代碼
/** * Definition of TreeNode: * public class TreeNode { * public int val; * public TreeNode left, right; * public TreeNode(int val) { * this.val = val; * this.left = this.right = null; * } * } */ public class Solution { /* * @param root: the root of the binary tree * @return: all root-to-leaf paths */ public List<String> binaryTreePaths(TreeNode root) { // write your code here List<List<Integer>> ans = new ArrayList<List<Integer>>(); List<Integer> temp = new ArrayList<Integer>(); List<Integer> temptemp; List<String> ansrel = new ArrayList<String>(); if(root == null){ return ansrel; } work(root,ans,temp); for(int i = 0; i<ans.size(); i++){ temptemp = ans.get(i); String tmp = ""; for(int j = 0; j<temptemp.size(); j++){ if(j == temptemp.size()-1){ tmp = tmp+temptemp.get(j)+""; }else{ tmp = tmp+temptemp.get(j)+"->"; } } ansrel.add(tmp); } return ansrel; } public void work(TreeNode root,List<List<Integer>> ans,List<Integer> temp){ if(root.left == null&&root.right == null){ temp.add(root.val); ans.add(new ArrayList<Integer>(temp)); temp.remove(temp.size()-1); }else if(root.left == null){ temp.add(root.val); work(root.right,ans,temp); temp.remove(temp.size()-1); }else if(root.right == null){ temp.add(root.val); work(root.left,ans,temp); temp.remove(temp.size()-1); }else{ temp.add(root.val); work(root.left,ans,temp); work(root.right,ans,temp); temp.remove(temp.size()-1); } return; } }
這道題還算簡單。 每次一層放入當前節點 然后左右子樹遞歸調用,然后當前層返回之前把當前層的值remove掉。
存進去的條件是沒有左右子樹。(如果插入條件寫成為空的話,會出現重復的情況,會把一個節點的左右空子樹放進去,各自生成一個,但是是一樣的)
最后給轉化成那種字符串形式。 注意這里的列表是索引傳遞 需要ans.add(new ArrayList(temp));
這樣寫
