7-59 还原二叉树 (25分)

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

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

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

输入样例:
9
ABDFGHIEC
FDHGIBEAC

输出样例:
5

#include <bits/stdc++.h>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof a)
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
char pre[55],in[55];
int n;
typedef struct node
{
	int v;
	struct node *l,*r;
}node,*tree;
tree create(char *in,char *pre,int len)
{
	if(len==0) return nullptr;
	tree t=new node({pre[0],nullptr,nullptr});
	int i=0;
	while(in[i]!=pre[0]) i++;
	t->l=create(in,pre+1,i);
	t->r=create(in+i+1,pre+i+1,len-i-1);
	return t;
}
int height(tree t)
{
	int th,rh,lh;
	if(!t) th=0;
	else
	{
		lh=height(t->l);
		rh=height(t->r);
		th=max(lh,rh);
		th++;
	}
	return th;
}
int main()
{
//	freopen("D:\\LYJ.txt","r",stdin);
	cin>>n;
	for(int i=0;i<n;i++) cin>>pre[i];
	for(int i=0;i<n;i++) cin>>in[i];
	tree T=create(in,pre,n);
	cout<<height(T);
	return 0;
}