（先序和中序）还原二叉树+求树的高度

给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。

输入格式:

输入首先给出正整数N（≤50），为树中结点总数。下面两行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母（区别大小写）的字符串。

输出格式:

输出为一个整数，即该二叉树的高度。

输入样例:

9

ABDFGHIEC
FDHGIBEAC

输出样例:

5

#include <bits/stdc++.h>
using namespace std;
typedef struct Node{
   char data;
    struct Node* lchild;
   struct Node* rchild;
}* Tree;
//恢复
Tree  firOrder(char fir[],char mid[],int len){

    if (len==0)
        return NULL;
    Tree bt=(Tree)malloc(sizeof(struct Node));
    bt->data=fir[0];
    int i=0;
    for(i=0;i<len;i++)
    {
        if(fir[0]==mid[i])//在中序中找到根节点得位置，分为左右子树
            break;
    }
    //分成左右两部分找，根据中序更新len得长度
    //fir[0]永远放当前树得根,左子树不用更新mid（因为是左开始，len已经更新le）
    bt->lchild=firOrder(fir+1,mid,i);
    //右侧要更新mid除去左侧和上次根得
    bt->rchild=firOrder(fir+i+1,mid+i+1,len-i-1);
    return bt;
}

int height(Tree bt){

    if(bt==NULL)
        return 0;
    int left=height(bt->lchild);
    int right=height(bt->rchild);
    return max(left,right)+1;//求高度每次左右子树得最大值+1
    
}
int main(){
    int n;
    cin>>n;
    getchar();//别忘了getchar()...
    
    char fir[n];
    for(int i=0;i<n;i++)
      cin>>fir[i];
    getchar();
    
   char mid[n];
    for(int i=0;i<n;i++)
        cin>>mid[i];
    
       Tree bt=firOrder(fir,mid,n);
    cout<<height(bt);

    return 0;
}