PTA-团体程序设计天梯赛-L2-004 这是二叉搜索树吗？ (25分)

L2-004 这是二叉搜索树吗？ (25分)

一棵二叉搜索树可被递归地定义为具有下列性质的二叉树：对于任一结点，

其左子树中所有结点的键值小于该结点的键值；
其右子树中所有结点的键值大于等于该结点的键值；
其左右子树都是二叉搜索树。

所谓二叉搜索树的“镜像”，即将所有结点的左右子树对换位置后所得到的树。

给定一个整数键值序列，现请你编写程序，判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。
输入格式：

输入的第一行给出正整数 N（≤1000）。随后一行给出 N 个整数键值，其间以空格分隔。
输出格式：

如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果，则首先在一行中输出 YES ，然后在下一行输出该树后序遍历的结果。数字间有 1 个空格，一行的首尾不得有多余空格。若答案是否，则输出 NO。
输入样例 1：

7
8 6 5 7 10 8 11

输出样例 1：

YES
5 7 6 8 11 10 8

输入样例 2：

7
8 10 11 8 6 7 5

输出样例 2：

YES
11 8 10 7 5 6 8

输入样例 3：

7
8 6 8 5 10 9 11

输出样例 3：

NO

思路：考察树的基本操作：建树，先序后序排列。

#include <bits/stdc++.h>
using namespace std;
struct Node
{
    int data,l,r;
}node[1005];
int seq=0;
void insert(int &root,int x)
{
    if(root==-1)
    {
        node[seq].data=x;
        node[seq].l=-1;
        node[seq].r=-1;
        root=seq++;
        return;
    }

        if(x<node[root].data)
            insert(node[root].l,x);
        else
            insert(node[root].r,x);


}
vector<int>vpre,vprem,vpost,vpostm;
void pre(int root,vector<int>&v)
{
    if(root==-1)
        return;
    v.push_back(node[root].data);
    pre(node[root].l,v);
    pre(node[root].r,v);
}
void prem(int root,vector<int>&v)
{
    if(root==-1)
        return;
    v.push_back(node[root].data);
    prem(node[root].r,v);
    prem(node[root].l,v);
}
void post(int root,vector<int>&v)
{
    if(root==-1)
        return;

    post(node[root].l,v);
    post(node[root].r,v);
    v.push_back(node[root].data);
}
void postm(int root,vector<int>&v)
{
    if(root==-1)
        return;

    postm(node[root].r,v);
    postm(node[root].l,v);
    v.push_back(node[root].data);
}

int main()
{
    int n; cin>>n;
    vector<int> a(n);

    for(int i=0;i<n;i++)
        cin>>a[i];
    int root=-1;
    for(int i=0;i<n;i++)
    {
        insert(root,a[i]);
    }
    pre(root,vpre);
    prem(root,vprem);
    post(root,vpost);
    postm(root,vpostm);
    if(vpre==a)
    {
        cout<<"YES\n";
        cout<<vpost[0];
        for(int i=1;i<n;i++)
            cout<<" "<<vpost[i];
        cout<<endl;
    }
    else if(vprem==a)
    {
        cout<<"YES\n";
        cout<<vpostm[0];
        for(int i=1;i<n;i++)
            cout<<" "<<vpostm[i];
        cout<<endl;
    }
    else
        cout<<"NO";

    return 0;
}