题目大意:
给定二叉树的后根序列和中根序列,输出层次序列。
输入:
第一行:节点个数
第二行:后根序列
第三行:中根序列
输出:
层次序列,数字间用一个空格隔开,末尾不允许有多余空格。
代码:
#include <stdio.h>
#include <queue>
using namespace std;
const int maxn=40;
int post[maxn],in[maxn];
int N;
struct biTreeNode{
int data;
biTreeNode* lchild;
biTreeNode* rchild;
};
biTreeNode* create(int postL,int postR,int inL,int inR){
if(postL>postR){
return NULL;//该结点为空指针,即没有结点
}
biTreeNode * root=new biTreeNode;//定义结构体指针变量需要new
root->data=post[postR];
int i;
for(i=inL;i<=inR;i++){
if(in[i]==post[postR]){
break;
}
}
//这里一定要小心,一定需要算出来左子树的结点个数,因为后根序列和中根的坐标后面是错开的
int k=i-inL;
root->lchild=create(postL,postL+k-1,inL,inL+k-1);
root->rchild=create(postL+k,postR-1,inL+k+1,inR);
return root;
}
int num=0;//计算输出的结点个数,用来控制空格输出,使得最后一个数字后没有空格
void level(biTreeNode * root){
queue<biTreeNode*> q;
q.push(root);
while(!q.empty()){
biTreeNode* now=q.front();
q.pop();
printf("%d",now->data);
num++;
if(num<N) printf(" ");
if(now->lchild!=NULL) q.push(now->lchild);
if(now->rchild!=NULL) q.push(now->rchild);
}
}
int main(){
scanf("%d",&N);
for(int i=0;i<N;i++){
scanf("%d",&post[i]);
}
for(int j=0;j<N;j++){
scanf("%d",&in[j]);
}
biTreeNode* root=create(0,N-1,0,N-1);
level(root);
return 0;
}