L2-006 树的遍历
分数 25
作者 陈越
单位 浙江大学
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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 tree{
int data;
struct tree* left;
struct tree* right;
}tree,*link;
link creat(int back[],int inorder[],int n);
void ceng(link root);
int main(){
int n;
int back[55];
int inorder[55];
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&back[i]);
}
for(int i=0;i<n;i++){
scanf("%d",&inorder[i]);
}
link root =( link)malloc(sizeof(tree));
root=creat(back,inorder,n);
ceng(root);
}
link creat(int back[],int inorder[],int n){
if(n==0){
return NULL;
}
link root= (link)malloc(sizeof(tree));
int index;//根节点在中序遍历中的位置
for(int i=0;i<n;i++){
if(back[n-1]==inorder[i]){
index=i;
break;
}
}
root->data=back[n-1];
// printf("%d index:%d\n",root->data,index);
root->left=creat(back,inorder,index);
root->right=creat(back+index,inorder+index+1,n-index-1); //注意下标变化
// 左 根 右
//左 右 根
//左节点是一样的,个数都是index个,所以右边子树的时候,
//后续遍历序列 中刨除 左子树的节点,剩下的便是右边子树的
return root;
}
void ceng(link root){
link queue[55];//数组模拟队列,其中元素存的是指针结点。
int head =0;//队头 ,出去
int rear=0; //队尾,进入
queue[rear++]=root;//录入根节点
int c=0;//用于保证输入格式
while(rear>head){//队列不为空
int start=head;//记录遍历开始的位置,即为本层的头
head=rear;//本层的尾巴rear即将成为下层的头。因为所有元素都弹出队列
//在这里下层的头head同时作为遍历结束的位置,因为在遍历中rear会不断改变,成为下层的尾
for(int i=start;i<head;i++){//弹出start到未变化的rear(即为head)之间的所有元素
if(c==0){
printf("%d",queue[i]->data);
c++;
}
else{
printf(" %d",queue[i]->data);
}
if(queue[i]->left!=NULL){//将该结点的左子树入队
queue[rear++]=queue[i]->left;
}
if(queue[i]->right!=NULL){//将该结点的右子树入队
queue[rear++]=queue[i]->right;
}
}
}
}