1020 Tree Traversals (25 分)
假设二叉树中的所有值都是不同的正整数。给定后序和中序遍历序列,输出相应二叉树的层序遍历序列。
输入规范
每个输入文件包含一个测试用例。对于每种情况,第一行给出一个正整数N(≤30),即二叉树中的节点总数。第二行给出后序,第三行给出了中序。一行中的所有数字都用空格隔开。
输出规范
对于每个测试用例,在一行中打印相应二叉树的层序遍历序列。一行中的所有数字必须用一个空格隔开,并且在行尾不能有多余的空格。
思路:与已知先序序列、中序序列,求二叉树一样。
代码
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn = 50;
struct node {
int data;
node* lchild;
node* rchild;
};
int in[maxn], post[maxn];
int n; //结点数
//先由已知求出树
node* creat ( int postL, int postR, int inL, int inR ){
if( postL > postR ){
return NULL;
}
node* root = new node;
root->data = post[postR];
int k;
for ( k=inL; k<=inR; k++ ){
if( in[k] == post[postR] ){
break;
}
}
int numLeft = k-inL;
root->lchild = creat ( postL, postL+numLeft-1, inL, k-1 );
root->rchild = creat ( postL+numLeft, postR-1, k+1, inR );
return root;
}
int num = 0; //记录已输出结点个数
//层序遍历输出
void BFS ( node* root ){
queue <node*> q;
q.push(root);
while ( !q.empty() ){
node* 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 i=0; i<n; i++ ){
scanf("%d", &in[i]);
}
node* root = creat ( 0, n-1, 0, n-1 );
BFS ( root );
return 0;
}