问题描述
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入数据有多组,每组数据第一行输入1个正整数N(1 <= N <=50)为树中结点总数,随后2行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区分大小写)的字符串。
输出一个整数,即该二叉树的高度。
看了下网上的答案大多数都是用指针和结构体来建立一棵树然后递归求解,但是其实也可以用数组来模拟二叉树来求得二叉树的高度。
先序遍历的顺序是 根结点 —> 左子树 —> 右子树
中序遍历的顺序是 左子树 —> 根结点 —> 右子树
所以我们可以遍历一遍先序遍历的结点,然后到中序遍历里面划分出左子树和右子树
代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std ;
int main()
{
// num为结点数目,ans为中序遍历中对应结点的高度
string pre , order ;
int i , j , ans[100] , right , left , num , x ;
while( cin >> num ){
// 初始化数组
memset(ans,0,sizeof(ans));
cin >> pre >> order ;
for( j = 0 ; j < num ; ++j ){
// 这里是把结点的字母的值映射到ans里面去,加上100是因为结点最多只有50个,所以高度最大为50,这就可以根据ans[i]的值是否大于100来判断这个结点的高度是否求出了,ans数组从0开始映射是为了在映射两边都设置个0
ans[j+1] = order[j] + 100 ;
}
for( i = 0 ; i < num ; ++i ){
for( j = 0 ; j < num ; ++j ){
if( pre[i] == order[j] ){
// 寻找右边已经求出高度的结点
for( x = j+2 ; x < num+2 ; ++x ){
if( ans[x] < 100 ){
right = ans[x] ;
break ;
}
}
// 寻找左边已经求出高度的结点
for( x = j ; x >= 0 ; --x ){
if( ans[x] < 100 ){
left = ans[x] ;
break ;
}
}
// 新找的这个结点是它左右已经求出高度的结点中较大的那个结点的孩子
ans[j+1] = right > left ? (right+1) : (left+1) ;
}
}
}
// 对各个结点的高度进行排序,输出最大高度
sort( ans+1, ans+1+num );
cout << ans[num] << endl ;
}
return 0 ;
}
为什么我不首先就考虑建立一棵树来递归求解呢?因为我数据结构没学好,对于建树操作不熟悉= = ,就想别的方法了
数据结构就要考试了,希望别挂科hhh