PTA 7-10 树的遍历(25 分)

时间:2022-01-20 12:26:44

给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式:

输入第一行给出一个正整数N),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式:

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。

输入样例:

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例:

4 1 6 3 5 7 2

     首先要先了解二叉树的遍历:https://www.cnblogs.com/liuamin/p/6269950.html


代码:
#include<stdio.h>
#include<iostream>
#include<stdlib.h>
using namespace std;


struct node{
int data;
node *lchild,*rchild;
};


int n;
int btree[100]={0};
int ctree[100]={0};


node * creatTree(int btree[],int ctree[],int n){
if(n<=0){
return NULL;
}
node *T;
//首结点 
int res = btree[n-1];         
T=(node*)malloc(sizeof(node));
T->data=res;
int j = 0;
for(int i = 0;i<n;++i){
if(ctree[i]==res){
j = i;
break;
}
}
T->lchild = creatTree(btree,ctree,j);
T->rchild = creatTree(btree+j,ctree+j+1,n-j-1);
return T;
}


void print(node *T){
node *p;
node *q[110];
//rear是数组的长度,front输出的位置 
int rear = 0,front = -1;
q[rear] = T;
int num = 0;
while(rear != front){
front++;
p = q[front];
cout << p->data;
num++;      //输出的次数
if(num<n){
cout << " ";

if(p->lchild!=NULL){
rear++;
q[rear]=p->lchild;
}
if(p->rchild!=NULL){
rear++;
q[rear]=p->rchild;
}
}
}


int main(){

cin >> n;
//后序遍历 
for(int i = 0;i<n;++i){
cin >> btree[i];
}
//中序遍历 
for(int j = 0;j<n;++j){
cin >> ctree[j];
}
node *T = creatTree(btree,ctree,n);
print(T);
return 0;
}