给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(≤30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例:
4 1 6 3 5 7 2
#include<stdio.h>
#include<stdlib.h>
typedef struct BiTNode {
char data;
struct BiTNode *lchild, *rchild;
} BiTNode, *BiTree;
typedef struct { //定义顺序队列
int data[40] = {0};
int front;
int rear;
} SqQueue;
int i=0;
int s[40]={0};
//post为后序遍历序列,mid为中序遍历序列。
//pl为当前子树序列的后序序列左边界,pr为右边界。同理,ml为当前子树序列的中序左边界,mr为右边界
void dispose(BiTree &T, int post[], int mid[], int pl, int pr, int ml, int mr) {
T = (BiTNode *)malloc(sizeof(BiTNode));
T->data = post[pr];
T->lchild = T->rchild = NULL;
int pos = ml;
while (mid[pos] != post[pr])
pos++;
int lenthL = pos - ml;//左子树的长度
if (pos > ml) //左子树
dispose(T->lchild, post, mid, pl, pl + lenthL - 1, ml, pos - 1);
//左子树,中缀的结尾移动到中缀中结点的左边,后缀的结尾移动到开头+lenthL-1处
if (pos < mr) //右子树
dispose(T->rchild, post, mid, pl + lenthL, pr - 1, pos + 1, mr);
}
int createTree(BiTree &T) {
int post[40] = {0};
int mid[40] = {0};
int length = 0;
scanf("%d", &length);
for (int i = 0; i < length; i++)
scanf("%d", &post[i]);
for (int i = 0; i < length; i++)
scanf("%d", &mid[i]);
dispose(T, post, mid, 0, length - 1, 0, length - 1);
return length;
}
void position(BiTree T, BiTNode * &P,int a) {
if(T!=NULL){
if(T->data==a){
P=T;
return ;
}
position(T->lchild,P,a);
position(T->rchild,P,a);
}
}
void levelOrderTraverse(BiTree T, SqQueue &Q) {
if (Q.front != Q.rear) { //若队列非空
int temp = Q.data[++Q.front];
s[i++]=temp;
BiTNode *P;
position(T,P,temp);
if (P->lchild != NULL)
Q.data[++Q.rear] = P->lchild->data;
if (P->rchild != NULL)
Q.data[++Q.rear] = P->rchild->data;
levelOrderTraverse(T, Q);
} else return;
}
int main() {
BiTree T;
int j = 0;
int length = createTree(T);
SqQueue Q;
Q.front = Q.rear = -1; //初始化
Q.data[++Q.rear] = T->data;
levelOrderTraverse(T, Q);
for(i=0;i<length-1;i++)
printf("%d ",s[i]);
printf("%d",s[i]);
return 0;
}