Problem Description
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
Input
输入数据有多组,每组数据第一行输入1个正整数N(1 <= N <= 50)为树中结点总数,随后2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区分大小写)的字符串。
Output
输出一个整数,即该二叉树的高度。
Sample Input
9
ABDFGHIEC
FDHGIBEAC
Sample Output
5
Hint
先序加中序建立二叉树的关键点在于明确先,中序中根节点与子节点之间的关系;
在先序中先遍历根节点,而在中序的根节点在左(或者右)节点之后遍历;
因此,在中序的字符串中找到根节点后,左侧的就是根节点的左子树,右侧为右子树(注:假设遍历方式为根,左,右)
#include <iostream>
using namespace std;
char s1[55],s2[55];
int n;
struct node///有的时候为了简单可能会加上typedef,但可有可无
{
char data;
node *lchild,*rchild;
};
node *creat(char s1[],char s2[],int len)
{
if(len<=0)return NULL;///长度不够是说明节点为空
node *root=new(node);
root->data=s1[0];
int i;
for(i=0;i<len;i++)
{
if(s2[i]==s1[0])break;///找到根节点在中序里的位置
}
root->lchild=creat(s1+1,s2,i);看样例体会下范围
root->rchild=creat(s1+1+i,s2+i+1,len-i-1);
return root;
}
int high(node *root)
{
if(!root)return 1;///此处的!root和root->lchild!=NULL&&root->rchild!=NULL不一样,前者是本身为空,后者是没有子节点,但是本身含有data数据
else return max(high(root->lchild),high(root->rchild))+1;
}
int main()
{
while(cin>>n)
{
cin>>s1>>s2;
node *T=creat(s1,s2,n);
cout<<high(T)-1<<endl;///多返回了一层,自己体悟
}
return 0;
}