4-14 还原二叉树 (15 分)
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
#include <bits/stdc++.h>
using namespace std;
string s1,s2;
int height(int t1,int t2,int node,int H)
{
int len=0;
if(node==0)return H-1;
while(1)
{
if(s1[t1]==s2[t2+len])
break;
len++;
}
int h1=height(t1+1,t2,len,H+1);
//cout << h1;
int h2=height(t1+len+1,t2+len+1,node-len-1,H+1);
//cout << h2;
return h1>h2?h1:h2;
}
int main()
{
int n;
cin >> n;
cin >> s1 >> s2;
//cout << s1 << s2;
cout << height(0,0,n,1);
return 0;
}