PTA天梯赛 7-59 还原二叉树 (25 分)

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

输入格式:

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

输出格式:

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

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

#include<bits/stdc++.h>
#include<string>
using namespace std;
struct node{
	int l;
	int r;
}tree[100];
int pre[60];
int mid[60]; 
int N;
int maxn=0;
int built(int a,int b,int c,int d,int deaph)//中，前
{
	if(a>b)
	return 0;
	int rt=pre[c];
	maxn=max(maxn,deaph);
	int p1=a;
	while(mid[p1]!=rt)
	++p1;
	int p2=p1-a;
	tree[rt].l=built(a,p1-1,c+1,p2+c,deaph+1);//构建左子树
	tree[rt].r=built(p1+1,b,p2+c+1,d,deaph+1);//构建右子树
	return rt;
}
int main(void)
{
	cin>>N;
	char c;
	int d=1;
	getchar();
	map<char,int>mp;//利用map给每个字母赋不同的值
	for(int i=1;i<=N;++i)
	{
		cin>>c;
		mp[c]=d++;
		pre[i]=mp[c];
	}
	getchar();
	for(int i=1;i<=N;++i)
	{
		cin>>c;
		mid[i]=mp[c];
	}
	built(1,N,1,N,1);//根节点深度为1
	cout<<maxn;
	return 0;
}