数据结构实验之二叉树四:还原二叉树
Time Limit: 1000MS
Memory Limit: 65536KB
Problem Description
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
Input
输入数据有多组,每组数据第一行输入
1
个正整数
N(1 <= N <= 50)
为树中结点总数,随后
2
行先后给出先序和中序遍历序列,均是长度为
N
的不包含重复英文字母
(
区分大小写
)
的字符串。
Output
输出一个整数,即该二叉树的高度。
Example Input
9 ABDFGHIEC FDHGIBEAC
Example Output
5
给你先序和中序建树。
首先你要明白的是
先序遍历遍历顺序:根节点,左子树,右子树。
中序遍历遍历顺序:左子树,根节点,右子树。
要想的出高度,可以先建树,然后求出最深的叶子节点的深度即可。
建树:
1.因为先序遍历先遍历的是根节点,所以我们首先找到先序的第一个字符,这个就是根节点。示例来说就是A。
2.然后在中序遍历中找到这个根节点。那么中序的左半部分就是左子树了,右半部分就是右子树惹。
3.递归相应的左子树和右子树即可。
注意:
用一个长度变量记录当前树的长度,用来结束递归。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN=50+5;
int l;
char in[MAXN],mid[MAXN];
struct node
{
char key;
struct node*left,*right;
};
struct node *creat()
{
struct node *p=new node;
p->left=p->right=NULL;
return p;
};
struct node*in_mid_build_tree(struct node *root,char *in,char *mid,int l)
{
if(!l)return 0;
root=creat();
root->key=*in;
int pos;
for(pos=0;pos<l;++pos)if(mid[pos]==*in)break;
root->left=in_mid_build_tree(root->left,in+1,mid,pos);
root->right=in_mid_build_tree(root->right,in+pos+1,mid+pos+1,l-pos-1);
return root;
};
int get_high(struct node *root,int high)
{
if(!root)return high-1;//空节点可不能算+1深度的
return max(get_high(root->left,high+1),get_high(root->right,high+1));
}
int main()
{
while(~scanf("%d",&l))
{
scanf("%s%s",in,mid);
struct node *root=in_mid_build_tree(root,in,mid,l);
printf("%d\n",get_high(root,1));
}
return 0;
}