给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
首先我们利用先序和中序来看一下如何确定一棵二叉树
1.在先序遍历中找到根节点
2.在中序遍历中找到该根节点所在的位置
3.中序遍历中该根节点的位置的左边就是该结点的左子树,右边就是右子树
4.将中序遍历的左右两部分继续执行前三个步骤,直到确定整个二叉树
比如利用这个图中所给的例子进行分析:
首先我们可以从先序遍历中确定A是整棵二叉树的根节点,然后我们在中序遍历中找到A的位置,A的左边就是他的左子树,右边就是它的右子树
接着我们从先序遍历中确定下一个根节点,为B
,在中序遍历中找到B,发现他在A的左子树中,所以B就是A的左子树的根节点
然后我们就可以画出相应的图
接着再去先序遍历中确定下一个根节点,下一个是D,D在中序遍历中在B的左边,说明D是B的左子树的根结点(同时可以确定E 为B 右子树的根结点)
在先序遍历中找下一个根节点F,对应中序遍历中,F为D的左边,所以是左子树的根结点
下一个是G,对应D的右边
接下来是H和I,可以在中序遍历中看到,H和I分别是G的左子树和右子树的根结点
后面的E和C很好确定,E是B的右子树根结点,C是A的右子树根结点
最后画出来的图
代码实现(C++)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
struct node
{
char data;
node* lchild;
node* rchild;
};
char pre[105],in[105];//pre是先序,in是中序
node* create(int preleft,int preright,int inleft,int inright)
{//先序遍历:[preleft,preright] 中序遍历:[inleft, inright]
if(preleft>preright)//如果左边界大于右边界
return NULL;
node* root=new node;//分配一个新的结点
root->data =pre[preleft];//先序遍历的第一个就是二叉树的根结点 ,将根结点存入
int k;
for(k=inleft;k<=inright;k++)//在中序遍历中找到根节点所在的位置
{
if(in[k]==pre[preleft])
break;
}
int numLeft=k-inleft;//左子树的结点个数
root->lchild =create(preleft+1,preleft+numLeft,inleft,k-1);//左子树的先序区间为[preleft+1,preleft+numLeft] ,中序区间为[inleft,k-1]
root->rchild =create(preleft+numLeft+1,preright,k+1,inright);//右子树的先序区间为[preleft+numLeft+1,preright] ,中序区间为[k+1,inright]
return root;
}
int depth(node *T)
{
if(T==NULL)
return 0;
int l=depth(T->lchild );
int r=depth(T->rchild );
return (l>r?l:r)+1;深度就是左右子树的最大值 +1
}
int main()
{
int n;
cin>>n;
getchar();
cin>>pre;
int len1=strlen(pre);
cin>>in;
int len2=strlen(in);
node* T=create(0,len1-1,0,len2-1);
cout<<depth(T);
return 0;
}