给定非空二叉树的中根序列和后根序列,请编写程序创建该二叉树,计算其高度和先根序列,最后删除该二叉树;如给定的中根和后根序列不合法,则亦能识别。
输入格式:
输入包含多组数据(不超过10组),每组为两行字符串,第一行表示某二叉树的后根序列,第二行表示其中根序列。结点的值均为A-Z的大写字母,故二叉树结点个数不超过26,且保证输入的两个序列都是结点的全排列,但不一定是合法的中根和后根序列。输入保证不是空二叉树。
输出格式:
对于每组数据,如果输入的序列不合法(不是同一棵树的中根序列和后根序列),则输出INVALID;若输入序列合法,输出为两行,第一行为一个整数,表示该二叉树的高度,第二行为一个字符串,表示该二叉树的先根序列。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
map<char,int> mp;
int pos;
typedef struct Node{
char val;
struct Node* l;
struct Node* r;
Node(char x):val(x),l(NULL),r(NULL){};
}TNode,*Tree;
string zhong,hou;
Tree build(int l,int r)
{
if(r<l){
return NULL;
}
char zhi=hou[pos];
pos--;
int index=mp[zhi];
Tree T=new Node(zhi);
T->r=build(index+1,r);
T->l=build(l,index-1);
return T;
}
bool judge(string zhong,string hou)//判断给出的序列是否合法
{
if(zhong.size()==0&&hou.size()==0) return true;//都是为0,合法
if(zhong.size()!=hou.size()) return false;//长度不一样肯定不合法
char ch=hou[hou.size()-1];//取出后序数组的那个字母
int index=0;
for(int i=0;i<zhong.size();i++){ //找ch在中序数组中所在的位置
if(ch==zhong[i]){
index=i;
break;
}
}
string lh=hou.substr(0,index);//后序数组的左部分
string rh=hou.substr(index,hou.size()-index-1);//后序数组的右部分
string lz=zhong.substr(0,index);//中序数组的左部分
string rz=zhong.substr(index+1);//中序数组的右部分
for(int i=0;i<lh.size();i++){//左部分的数组,元素必须要一致
if(lz.find(lh[i])==-1) return false;
}
for(int i=0;i<rh.size();i++){//右部分的数组,元素必须要一致
if(rz.find(rh[i])==-1) return false;
}
return judge(lz,lh)&&judge(rz,rh);//继续分割左和右
}
int getH(Tree T)
{
if(!T) return 0;//走到空1,就为 0