codeup模拟赛 进击的二叉查找数

问题 B: 进击的二叉查找树

时间限制: 1 Sec

内存限制: 64 MB

提交: 1017

解决: 379

提交状态

题目描述

给定1~N的两个排列，使用这两个排列分别构建两棵二叉查找树（也就是通过往一棵空树中依次插入序列元素的构建方式）。如果这两棵二叉查找树完全相同，那么输出YES；否则输出NO。之后，输出第一个排列对应的二叉查找树的后序序列、层序序列。

输入

每个输入文件中一组数据。

第一行1个正整数N（1<=N<=30），表示二叉查找树中的结点个数。

接下来两行，代表1~N的两个排列。

输出

如果两个排列代表的二叉查找树完全相同，那么输出一行YES，否则输出一行NO。

接下来两行分别输出第一个排列对应的二叉查找树的后序序列、层序序列，整数之间用空格隔开。

每行末尾不允许有多余的空格。

样例输入

5
4 2 1 3 5
4 5 2 3 1

样例输出

YES
1 3 2 5 4
4 2 5 1 3

代码：

#include<cstdio>#include<cstdlib>#include<iostream>#include<string>#include<cstring>#include<algorithm>#include<queue>using namespace std;typedef struct BiTNode{    int data;    struct BiTNode *lchild, *rchild;}BiTNode, *BiTree;int pos=0;int BSTInsert(BiTree *t, int element){    if(NULL==*t)    {        (*t)=(BiTree)malloc(sizeof(BiTNode));        (*t)->data=element;        (*t)->lchild=(*t)->rchild=NULL;        return 1;    }    if(element==(*t)->data)        return 0;    if(element<(*t)->data)        return BSTInsert(&(*t)->lchild,element);    return BSTInsert(&(*t)->rchild,element);}void CreateBST(BiTree *t, int *a, int n){    (*t) = NULL;    for( int i=0; i<n; i++ )        BSTInsert(t,a[i]);}void PrintBST(BiTree t, int a[]){    if(t)    {        PrintBST(t->lchild,a);        PrintBST(t->rchild,a);        a[pos]=t->data;        pos++;    }}bool isEqual(BiTree t1, BiTree t2){    if(t1==NULL&&t2==NULL)        return true;    else if(t1==NULL||t2==NULL)        return false;    else    {        if(t1->data!=t2->data)            return false;        else        {            bool isEqualLeft,isEqualRight;            isEqualLeft=isEqual(t1->lchild,t2->lchild);            isEqualRight=isEqual(t1->rchild,t2->rchild);            if(isEqualLeft&&isEqualRight)                return true;            else            {                isEqualLeft=isEqual(t1->lchild,t2->rchild);                isEqualRight=isEqual(t1->rchild,t2->lchild);                if(isEqualLeft&&isEqualRight)                    return true;                else                    return false;            }        }    }}void printLevel(BiTree t, int a[], int index){    queue<BiTree> q;    if(t!=NULL)        q.push(t);    BiTree b;    while(!q.empty())    {        b=q.front();        a[index]=b->data;        index++;        q.pop();        if(b->lchild)            q.push(b->lchild);        if(b->rchild)            q.push(b->rchild);    }}int main(){    int n;    int *a,*b;    BiTree t1,t2;    scanf("%d", &n);    a=(int*)malloc(sizeof(int)*n);    b=(int*)malloc(sizeof(int)*n);    for(int i=0;i<n;i++)        scanf("%d",&a[i]);    for(int i=0;i<n;i++)        scanf("%d",&b[i]);    CreateBST(&t1,a,n);    CreateBST(&t2,b,n);    if(isEqual(t1,t2))        printf("YES\n");    else        printf("NO\n");     PrintBST(t1,a);    printf("%d",a[0]);    for(int i=1;i<n;i++)    {        printf(" %d",a[i]);    }    printf("\n");    printLevel(t1,a,0);    printf("%d",a[0]);    for(int i=1;i<n;i++)    {        printf(" %d",a[i]);    }    printf("\n");    return 0;}