题目描述
现给定一棵二叉树的先序遍历序列和中序遍历序列,要求你计算该二叉树的高度。
输入
输入包含多组测试数据,每组输入首先给出正整数N(<=50),为树中结点总数。下面2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出
对于每组输入,输出一个整数,即该二叉树的高度。
样例输入
9
ABDFGHIEC
FDHGIBEAC
7
Abcdefg
gfedcbA
样例输出
5
7
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int solve(char *a,char *b,int n)
{
if(n==0)return 0; //如果是空树,高度便为0
int i;
for(i=0;i<n;i++)
{
if(b[i]==a[0])break; //寻找根结点在中序中的位置下标
}
int left=solve(a+1,b,i); //pre[0]作为root后,后一个作为新的root
int right=solve(a+i+1,b+i+1,n-i-1); //i是根结点在中序中的位置下标,b+i+1即右子树且跳过根结点,n-i-1同样是跳过根结点,i+n-i-1=n-1
return max(left,right)+1; //root那一层,故高度+1
}
int main()
{
int n;
char a[80],b[80];
while(cin>>n)
{
cin>>a>>b;
int ans=solve(a,b,n);
cout<<ans<<endl;
}
return 0;
}
利用二叉树的特性,采用递归函数;通过改变参数来实现遍历