二叉樹的所有路徑


給一棵二叉樹,找出從根節點到葉子節點的所有路徑。

樣例

給出下面這棵二叉樹:

   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));
這樣寫


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM