Problem Description
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
Input
输入数据有多组,每组数据第一行输入1个正整数N(1 <= N <= 50)为树中结点总数,随后2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区分大小写)的字符串。
Output
输出一个整数,即该二叉树的高度。
Sample Input
9
ABDFGHIEC
FDHGIBEAC
Sample Output
5
Hint
Source
xam
题解:最底层叶子节点高度为1,依次往上迭代即可
#include<bits/stdc++.h>
using namespace std;
struct node
{
node *l,*r;
char data;
};
node *create(char *previous,char *middle,int len)
{
if(len<=0)
return NULL; //终止条件很重要
node *root;
root=new node;
root->data=previous[0];
int i;
for(i=0;i<len;i++)
{
if(middle[i]==previous[0])//找到中序遍历中根节点的分界
break;
}
root->l=create(previous+1,middle,i);//画个图很好理解
root->r=create(previous+i+1,middle+i+1,len-1-i);
return root;
}
int getheight(node *root)
{
int h=0; //最底层高度为0,则在叶子节点中返回高度为1,依次往上迭代
if(root)
h=max(getheight(root->l),getheight(root->r))+1;
return h;
}
int main()
{
int n;
char previous[100],middle[100];
while(cin>>n)
{
cin>>previous>>middle;
node *root;
root=create(previous,middle,n);
int ans=getheight(root);
cout<<ans<<endl;
}
return 0;
}
/***************************************************
User name: ACM18171信科1801张林
Result: Accepted
Take time: 0ms
Take Memory: 192KB
Submit time: 2019-02-25 13:14:30
****************************************************/