给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
根据前序序列和中序序列来确定左右子树,先找到树根,中序序列的树根前面的字符串是其左子树,后面是右子树。再将左子树和右子树的序列分别作为一棵树进入递归,对其重复找出树根和左右子树的处理,每次记录下当前左右子树的深度,最后返回最大深度。
#include<bits/stdc++.h>
using namespace std;
int dfs(char a[],char b[],int n){
int i;
if(n==0)return 0;
for(i=0;i<n;i++){
if(b[i]==a[0]){
break;
}
}
int x=dfs(a+1,b,i)+1;//求左子树的深度
int y=dfs(a+i+1,b+i+1,n-i-1)+1;//求右子树的深度
return x>y?x:y;
}
int main()
{
char a[101];
char b[101];
int n;
cin>>n;
cin>>a>>b;
int cnt=dfs(a,b,n);
cout<<cnt<<endl;
}