【PAT天梯赛】【数据结构之简单数据结构】树和二叉树，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 <iostream>
#include <vector>
using namespace std;
const int maxn = 505;
#define mem(a) memset(a,0,sizeof(a))
#define inf 0x3f3f3f3f
struct node{
    int data;
    node *lchild,*rchild;
};
void create(node* &T,int data)//建树
{
   if(T == NULL)
   {
       T = new node;
       T->data=data;
       T->lchild = T->rchild = NULL;
       return ;
   }
   if(data < T->data)
       create(T->lchild,data);
   else
       create(T->rchild,data);
}
void preorder(node* T,vector<int> &vi)//前序遍历二叉树
{
    if(T == NULL)return;
    vi.push_back(T->data);
    preorder(T->lchild,vi);
    preorder(T->rchild,vi);
}
void pre_mirror(node* T,vector<int>&vi)//得到前序镜像二叉树遍历结果
{
    if(T == NULL)
        return;
    vi.push_back(T->data);
    pre_mirror(T->rchild,vi);
    pre_mirror(T->lchild,vi);
}
void postorder(node* T,vector<int> &vi)//后序遍历二叉树
{
    if(T == NULL)return;
    postorder(T->lchild,vi);
    postorder(T->rchild,vi);
    vi.push_back(T->data);
}
void post_mirror(node* T,vector<int>&vi)//得到后序镜像二叉树遍历结果
{
    if(T == NULL)
        return;
    post_mirror(T->rchild,vi);
    post_mirror(T->lchild,vi);
    vi.push_back(T->data);
}
vector<int> initial,pre,pre_m,post,post_m;
int main()
{
    int n,data;
    node* T = NULL;
    cin>>n;
    for(int  i = 0;i  <n;i++)
    {
        cin>>data;
        initial.push_back(data);
        create(T,data);//建立二叉搜索树
    }
    preorder(T,pre);
    pre_mirror(T,pre_m);
    postorder(T,post);
    post_mirror(T,post_m);
    if(initial == pre)//给定树与二叉搜索树的前序遍历或镜像的前序遍历相同，则为二叉搜索树
    {
        puts("YES");
        for(int i = 0;i < post.size();i++)
        {
            if(i)
                cout<<" ";
            cout<<post[i];
        }
    }
    else if(initial == pre_m)
    {
        puts("YES");
        for(int i = 0;i < post_m.size();i++)
        {
            if(i)
                cout<<" ";
            cout<<post_m[i];

        }
    }
    else
        puts("NO");

    return  0;
}