PAT L2-004. 这是二叉搜索树吗？

题目链接：PAT L2-004. 这是二叉搜索树吗？

题意：

给你一串数，问你是否是一个二叉搜索树或者是镜像的先序遍历。

如果是，就输出这棵树的后序遍历。

题解：

根据二叉搜索树的性质，模拟判断一下。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;

const int N=1e4+7;

int n,ismirror;
vector<int>pre,ans;

void dfs(int l,int r)
{
    if(r<l)return;
    int b=l,c=r+1;
    if(ismirror)
    {
        while(b<r&&pre[b+1]>=pre[l])b++;
        while(c>l+1&&pre[c-1]<pre[l])c--;
    }else
    {
        while(b<r&&pre[b+1]<pre[l])b++;
        while(c>l+1&&pre[c-1]>=pre[l])c--;
    }
    if(c-b!=1)return;
    dfs(l+1,b),dfs(c,r);
    ans.push_back(pre[l]);
}

int main()
{
    scanf("%d",&n);
    F(i,1,n)
    {
        int x;
        scanf("%d",&x);
        pre.push_back(x);
    }
    dfs(0,n-1);
    if(ans.size()!=n)ismirror=1,ans.clear(),dfs(0,n-1);
    if(ans.size()==n)
    {
        puts("YES");
        for(int i=0;i<n;i++)printf("%d%c",ans[i]," 
"[i==n-1]);
    }else puts("NO");
    return 0;
}