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不匹配时，计算过程会有很多次。
迭代代码如下