进击的二叉查找树

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;
}