1043. Is It a Binary Search Tree (25)--BST引用插入、二叉树遍历模板

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cmath>

#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>

using namespace std;

int N;
vector<int> v1,vpre,vmir;

struct node
{
    int data;
    node* l;
    node* r;
};

void insertt(int x,node* &p)//BST插入用引用
{
    if(p==NULL)
    {
        p=(node*)malloc(sizeof(node));
        p->data=x;
        p->l=NULL;
        p->r=NULL;
        return;
    }

    if(x < p->data)
    {
        insertt(x,p->l);
    }
    else if(x >= p->data)
    {
        insertt(x,p->r);
    }
}
void pre(node* p)//二叉树遍历模板
{
    if(p!=NULL)
    {
        vpre.push_back(p->data);
        pre(p->l);

        pre(p->r);
    }

}

void mir(node* p)
{
    if(p!=NULL)
    {
        vmir.push_back(p->data);
        mir(p->r);
        mir(p->l);
    }
}

bool judge1()
{
    for(int i=0;i<N;i++)
    {
        if(v1[i]!=vpre[i])
            return 0;
    }

    return 1;
}

bool judge2()
{
    for(int i=0;i<N;i++)
    {
        if(v1[i]!=vmir[i])
            return 0;
    }

    return 1;
}

int flag=0;
void tra(node* p)
{
    if(p!=NULL)
    {
        tra(p->l);
        tra(p->r);

        if(flag==1)
        {
            printf(" ");
            printf("%d",p->data);
        }
        else
        {
            flag=1;
            printf("%d",p->data);
        }
    }
}

void tra2(node* p)
{
    if(p!=NULL)
    {
        tra2(p->r);
        tra2(p->l);

        if(flag==1)
        {
            printf(" ");
            printf("%d",p->data);
        }
        else
        {
            flag=1;
            printf("%d",p->data);
        }
    }
}

int main()
{
   // freopen("in.txt","r",stdin);

    scanf("%d",&N);
    node* head=NULL;
    for(int i=0;i<N;i++)
    {
        int t;
        scanf("%d",&t);
        v1.push_back(t);
        insertt(t,head);
    }

    pre(head);
    mir(head);

    if(judge1()==1)
    {
        printf("YES\n");
        tra(head);
    }
    else if(judge2()==1)
    {
        printf("YES\n");
        tra2(head);
    }
    else
    {
        printf("NO");
    }



    return 0;
}