leetcode從先序遍歷還原二叉樹

題目鏈接：從先序遍歷還原二叉樹
題目描述：
我們從二叉樹的根節點 root 開始進行深度優先搜索。

在遍歷中的每個節點處，我們輸出 D 條短劃線（其中 D 是該節點的深度），然後輸出該節點的值。（如果節點的深度爲 D，則其直接子節點的深度爲 D + 1。根節點的深度爲 0）。

如果節點只有一個子節點，那麼保證該子節點爲左子節點。

給出遍歷輸出 S，還原樹並返回其根節點 root。

示例 1：

輸入：“1-2–3--4-5–6--7”
輸出：[1,2,5,3,4,6,7]
示例 2：

輸入：“1-2–3—4-5–6—7”
輸出：[1,2,5,3,null,6,null,4,null,7]
示例 3：

輸入：“1-401–349—90–88”
輸出：[1,401,null,349,88,90]

提示：

原始樹中的節點數介於 1 和 1000 之間。
每個節點的值介於 1 和 10 ^ 9 之間。

題目分析：對於給出的字符串S，分隔符是一個很好的入手點，根據分隔符能夠很快劃分出二叉樹的左右子樹
比如字符串：1-2–3—4-5–6—7

根據- 分爲 根1 左子樹 2--3--4 右子樹5--6--7
繼續根據-- 左左子樹分爲 根2 左子樹3 右子樹 4 
         右右子樹同理  根5 左6 右7
不難看出，這是個遞歸處理的過程，比較麻煩的點就是字符串的拆分了，拆分好之後遞歸處理即可
利用正則表達式進行處理
這裏注意一個點，我代碼中使用的方式有一個特點，當2--3--4，以--進行拆分時，會拆成2，3--4，
這樣2--3--4與2--3---4的這樣拆的話得到的結果是一樣的，那麼需要再次處理一下，看代碼吧

class Solution {
    public TreeNode recoverFromPreorder(String S) {
        TreeNode root = reviviTree(1, S);
        return root;
    }
     private TreeNode reviviTree(int layer,String str){
      if(str.equals("")) return null;
        String regex = "";
        //深度生成分隔符
        for(int i = 0;i < layer;i++) regex += '-';
        //根據分隔符拆分
        str=str.replaceAll("([0-9])"+regex+"([0-9])","$1,$2");
        String[] nodes = str.split(",");
        TreeNode root = new TreeNode(Integer.valueOf(nodes[0]));
        // System.out.println(str+" "+regex+" "+nodes.length);
        String leftStr = "",rightStr = "";
        //2，3--4形式需要再次處理
        if(nodes.length==2){
             nodes[1]=nodes[1].replaceAll("([0-9])"+regex+"([0-9])","$1,$2");
            String[] te = nodes[1].split(",");
            if(te.length>1){//左右均有
              leftStr = te[0];
              rightStr = te[1];
            }else{//只有左子樹
                leftStr = te[0];
                rightStr = "";
            }
           root.left = reviviTree(layer+1, leftStr);
           root.right = reviviTree(layer+1, rightStr);
        }
        //1，2--3--4，5--6--7形式直接傳遞即可
        if(nodes.length==3){
           root.left = reviviTree(layer+1, nodes[1]);
           root.right = reviviTree(layer+1, nodes[2]);
        }

        return root;
    }
}

其實思路挺簡單的就是一個遞歸，只不過字符串的處理麻煩了些。。。。。。天啊，給我一個萬能的工具吧