leetcode 1028. 從先序遍歷還原二叉樹

解法

1. 遞歸

涉及到樹的先序遍歷立刻就想到了遞歸，如下是遵照二叉樹先序遍歷順序寫的遞歸做法。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //直接在遞歸中解析字符串
    public TreeNode dfs(String s,int[] index,int deep){
        if(index[0]>=s.length())return null;
        //獲取當前深度。
        int hh=0,i=index[0],n=s.length();
        while (i<n&&s.charAt(i)=='-'){
            hh++;
            i++;
        }
        if(hh!=deep)return null;
        //獲取數據
        int dd=0;
        while(i<n&&s.charAt(i)!='-'){
            dd=dd*10+s.charAt(i)-'0';
            i++;
        }
        TreeNode node = new TreeNode(dd);
        index[0]=i;
        node.left = dfs(s,index,deep+1);
        node.right = dfs(s,index,deep+1);
        return node; 
    }

    public TreeNode recoverFromPreorder(String S) {
        return dfs(S,new int[]{0},0);
    }
}

TreeNode dfs(String s,int[] index,int deep)
其中，s爲傳入的字符串，index是字符串下一步被掃描的下標，由於java中沒法在函數中改變基本類型形參的值，所以用數組封裝一下。deep代表此時的深度，若在字符串中獲取到的深度（即讀取到的‘-’的個數）與deep一致，則表明接下來掃描的數據是此時遞歸的節點。

2. 迭代

遞歸的好處是簡單易懂，但是從效率上講應盡力避免遞歸，而所有的遞歸都可以用迭代實現。
上面的計算可以看到有很多冗餘之處，比如在計算深度時，在深度與deep不匹配時，計算過程會有很多次。
迭代代碼如下