7-22 还原二叉树 (25 分)
给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的高度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
主要就是用到了递归的方式来创建二叉树,要注意创建的时候,区间的选取和长度。
#include<iostream>
using namespace std;
typedef struct TreeNode *Position;
typedef Position BinTree;
struct TreeNode {
BinTree left;
BinTree right;
char cp;
};
BinTree findTree(char* in , char* pre , int len) {
if(len==0)
return NULL;//空的
BinTree str=new TreeNode;
str->cp=*pre;//根节点是先序的第一个
int temp=0;
for(int i=0; i<len; i++) {
if(in[i]==*pre)//在中序里找到根节点
break;
temp++;
}
str->left=findTree(in,pre+1,temp);//用前半部分来创建左子树
str->right=findTree(in+temp+1,pre+temp+1,len-temp-1);//用后半部分来创建右子树,长度还要减去根节点那个
return str;//返回根节点
}
int Len(BinTree str) {
if(str==NULL)
return 0;
int l=Len(str->left);//左子树
int r=Len(str->right);//右子树
return l>r?l+1:r+1;//还有根节点
}
int main() {
char in[10000],pre[10000];
int n;
cin>>n;
char cp;
for(int i=0; i<n; i++) {//先序
cin>>cp;
pre[i]=cp;
}
for(int i=0; i<n; i++) {//中序
cin>>cp;
in[i]=cp;
}
BinTree Node=new TreeNode;
Node=findTree(in,pre,n);
cout<<Len(Node)<<endl;
}