PAT A1043 二叉排序树的插入跟镜像树的前后序

PAT A1043 二叉排序树的插入跟镜像树的前后序

题目：给出n个正整数来作为一棵二叉排序树的结点插入顺序，问：这串序列是否是该二叉排序树的先序序列或是该二叉树的镜像的先序序列。所谓镜像树就是指交换二叉树的所有结点的左右子树而形成的树（也即左子树所有结点数据域大于或等于根结点，而根结点数据域小于右子树所有结点的数据域）。如果是镜像树，则输出yes,并输出对应的树的后序序列；否则，输出no

代码：

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
struct Bitnode{
    int data;
    struct Bitnode *lchild,*rchild;

}*Bitree;
void insert( Bitnode *&root,int data)
{
    if(root==NULL)
    {
        root=new  Bitnode;
        root->data=data;
        root->lchild=root->rchild=NULL;
        return ;

    }
    else if(data>root->data)
        return insert(root->rchild ,data);
    else
        return insert (root->lchild ,data);
}
void preOrder(Bitnode *root,vector <int>&vi)
{
    if(root==NULL) return;
    vi.push_back(root->data);
    preOrder(root->lchild,vi);
    preOrder(root->rchild,vi);

}   //先序遍历，结果存在VI
void preOrderMirror(Bitnode *root,vector<int>&vi)
{
    if(root==NULL)return;
    vi.push_back(root->data);
    preOrderMirror(root->rchild,vi);
    preOrderMirror(root->lchild,vi);
}//镜像树先序遍历，结果存放VI
void postOrder(Bitnode *root,vector <int>&vi)
{
    if(root==NULL) return;
    postOrder(root->lchild,vi);
    postOrder(root->rchild,vi);
    vi.push_back(root->data);


}   //后序遍历，结果存在VI
void postOrderMirror(Bitnode *root,vector<int>&vi)
{
    if(root==NULL)return;
    postOrderMirror(root->rchild,vi);
    postOrderMirror(root->lchild,vi);
    vi.push_back(root->data);

}//镜像树后序遍历，结果存放VI
vector <int> origin,pre,preM,post,postM;

int main() {
    int n,data;
    Bitnode *root=NULL;
    scanf("%d",&n);
    for(int i=0;i<n;i++) {
        scanf("%d",&data);
        origin.push_back(data);
        insert(root, data);
    }
    preOrder(root,pre);
    preOrderMirror(root,preM) ;
    postOrder(root,post);
    postOrderMirror(root,postM) ;
    if(origin==pre){
        printf("yes\n");
        for(int i=0;i<post.size();i++)
        {
            printf("%d",post[i]);
            if(i<post.size()-1)
                printf(" ");
        }
    }else if(origin==preM)
    {
        printf("yes\n");
        for(int i=0;i<post.size();i++)
        {
            printf("%d",postM[i]);
            if(i<postM.size()-1)
                printf(" ");
        }
    }else{
        printf("NO\n");
    }
    return 0;
}