LeetCode 1028 从先序遍历还原二叉树

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

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

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

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

思路：
先计算出每个节点是父亲（根）的第几代
然后再利用深度优先搜索，递归

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
 //先序遍历左边-右边
 //二叉树是根-左-右-根-左右
class Solution {
public:
    TreeNode* build(queue<int>&gen,queue<int>& num,int depth){//递归查找
        if(gen.front()!=depth)  return NULL;
        TreeNode* root=new TreeNode(num.front());
        gen.pop();
        num.pop();
        root->left=build(gen,num,depth+1);
        root->right=build(gen,num,depth+1);
        return root;
    }
    TreeNode* recoverFromPreorder(string S) {
        int left=0;
        queue<int> gen,num;
        for(int i=0;i<S.size();i++){
            if(S[i]!='-'){
                gen.push(i-left);
                left=i;
                while(i<S.size()&&S[i]!='-') i++;
                num.push(stoi(S.substr(left,i-left)));
                left=i;
            }
        }
        return build(gen,num,0);//gen保存的是第几代，num保存的是节点数
    }
};