Problem Description
给定1~N的两个排列,使用这两个排列分别构建两棵二叉查找树(也就是通过往一棵空树中依次插入序列元素的构建方式)。如果这两棵二叉查找树完全相同,那么输出YES;否则输出NO。之后,输出第一个排列对应的二叉查找树的后序序列、层序序列。
Input
每个输入文件中一组数据。
第一行1个正整数N(1<=N<=30),表示二叉查找树中的结点个数。
接下来两行,代表1~N的两个排列。
Output
如果两个排列代表的二叉查找树完全相同,那么输出一行YES,否则输出一行NO。
接下来两行分别输出第一个排列对应的二叉查找树的后序序列、层序序列,整数之间用空格隔开。
每行末尾不允许有多余的空格。
Sample Input
5
4 2 1 3 5
4 5 2 3 1
Sample Output
YES
1 3 2 5 4
4 2 5 1 3
Author
Shoutmon
Source
16浙大考研机试模拟赛
#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
struct TreeNode
{
int val;
TreeNode *left,*right;
TreeNode(int e):val(e),left(NULL),right(NULL){}
};
int n,e;
TreeNode * insert(TreeNode *r,int e)
{
if(r==NULL) return new TreeNode(e);
if(e<r->val) r->left=insert(r->left,e);
else r->right=insert(r->right,e);
return r;
}
bool issame(TreeNode *r1,TreeNode *r2)
{
if(!r1&&!r2) return true;
if(r1&&r2) return r1->val==r2->val&&issame(r1->left,r2->left)&&issame(r1->right,r2->right);
return false;
}
void postra(TreeNode *r,int &w)
{
if(r){
postra(r->left,w);
postra(r->right,w);
++w==n?printf("%d\n",r->val):printf("%d ",r->val);
}
}
void levtra(TreeNode *r)
{
queue<TreeNode *> q;
q.push(r);
int w=0;
while(!q.empty())
{
TreeNode *s=q.front();
q.pop();
++w==n?printf("%d\n",s->val):printf("%d ",s->val);
if(s->left) q.push(s->left);
if(s->right) q.push(s->right);
}
}
int main()
{
scanf("%d",&n);
TreeNode *r1=NULL,*r2=NULL;
for(int i=0;i<n;++i){
scanf("%d",&e);
r1=insert(r1,e);
}
for(int i=0;i<n;++i){
scanf("%d",&e);
r2=insert(r2,e);
}
bool res=issame(r1,r2);
res?printf("YES\n"):printf("NO\n");
int w=0;
postra(r1,w);
levtra(r1);
return 0;
}