L2-006. 树的遍历
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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 <bits/stdc++.h>
typedef struct node
{
int data;
struct node *l,*r;
} BTnode, *BiTree;
int flag, n;
BiTree Rstore(int b[],int c[],int n)
{
if(n==0)return NULL;
BiTree bt;
bt = (BTnode*)malloc(sizeof(BTnode));
bt->data = c[n-1];
int i;
for( i = 0; i < n; i++ )
{
if(b[i]==c[n-1])break;
}
bt->l = Rstore(b,c,i);
bt->r = Rstore(b+i+1,c+i,n-i-1);
return bt;
}
void levelorder(BiTree bt)
{
BiTree *q;
q = (BiTree *)malloc(sizeof(BiTree)*n);
int in,out;
if(bt==NULL)return;
in = 0;
out = 0;
q[in++] = bt;
while(in>out)
{
if(q[out])
{
if(!flag)
{
printf("%d",q[out]->data);
flag = 1;
}
else printf(" %d",q[out]->data);
q[in++] = q[out]->l;
q[in++] = q[out]->r;
}
out++;
}
}
int main()
{
BiTree T;
int *b, *c;
int i;
scanf("%d",&n);
b = (int *)malloc(sizeof(int)*n);
c = (int *)malloc(sizeof(int)*n);
for( i = 0; i < n; i++ )
scanf("%d",&b[i]);
for( i = 0; i < n; i++ )
scanf("%d",&c[i]);
flag = 0;
T = Rstore(c,b,n);
levelorder(T);
printf("\n");
return 0;
}