经典算法面试题：从先序遍历还原二叉树

题目

我们从二叉树的根节点 root 开始进行深度优先搜索。

在遍历中的每个节点处，我们输出 D 条短划线（其中 D 是该节点的深度），然后输出该节点的值。（如果节点的深度为 D，则其直接子节点的深度为 D + 1。根节点的深度为 0）。

如果节点只有一个子节点，那么保证该子节点为左子节点。

给出遍历输出 S，还原树并返回其根节点 root。

示例 1：

输入："1-2--3--4-5--6--7"
输出：[1,2,5,3,4,6,7]

思路

发现如果A为B的子节点，那么DepthA>DepthB。（‘-’个数）
所以我们只需要对S字符串进行全局遍历即可。同一个全局变量cur去处理节点值，同时用一个全局变量curD维护当前遍历到的深度，再用一个栈变量depth维护当前节点深度，如果curD>depth则显然全局遍历的下一节点为当前节点的子节点，否则为其长辈节点

实现

int cur = 0;
    int curD = 0;

    public TreeNode recoverFromPreorder(String S) {
        char[] nodes = S.toCharArray();
        return dfs(0,nodes);
    }

    public TreeNode dfs(int depth, char[] nodes){
        int val = 0;
        while (cur<nodes.length && nodes[cur]!='-') {
            val = val * 10 + nodes[cur] - '0';
            cur++;
        }

        curD = 0;

        while (cur<nodes.length && nodes[cur]=='-') {
            cur++;
            curD++;
        }

        TreeNode r = new TreeNode(val);

        if (curD > depth) {
            r.left = dfs(curD,nodes);
        }
        if (curD > depth) {
            r.right = dfs(curD,nodes);
        }

        return r;
    }