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<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-8
#define x first
#define y second
const int mod = 1e9 + 7;
const int MOD = 1e4+7;
const int N = 1e5 + 10;
const int M = 1111;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
int dxy[][2]={{0,1},{1,0},{1,1},{-1,1}};
using namespace std;

int dp[M];
int tot=0;

bool judge(int l,int r){
    int k=l+1;
    int tot=0;
    if(l>=r) return 1;
    rep(i,l+1,r){
        if(dp[i]>=dp[l]&&!tot){
            k=i;
            tot++;
        }
        else if(tot&&dp[i]<dp[l]) return 0;
    }
    if(judge(l+1,k-1)&&judge(k,r)) return 1;
    else return 0;
}

void post(int l,int r){
    int k=l+1;
    if(l>r) return ;
    else if(l==r){
        if(!tot){
            cout<<dp[l];
            tot++;
        }
        else cout<<" "<<dp[l];
        return ;
    }
    rep(i,l+1,r){
        if(dp[i]>=dp[l]){
            k=i;
            break;
        }
    }
    post(l+1,k-1);
    post(k,r);
    cout<<" "<<dp[l];
    return ;
}

bool check(int l,int r){
    int k=l+1;
    int tot=0;
    if(l>=r) return 1;
    rep(i,l+1,r){
        if(dp[i]<dp[l]&&!tot){
            k=i;
            tot++;
        }
        else if(tot&&dp[i]>=dp[l]) return 0;
    }
    if(check(l+1,k-1)&&check(k,r)) return 1;
    else return 0;
}

void dis(int l,int r){
    int k=l+1;
    if(l>r) return ;
    else if(l==r){
        if(!tot){
            cout<<dp[l];
            tot++;
        }
        else cout<<" "<<dp[l];
        return ;
    }
    rep(i,l+1,r){
        if(dp[i]<dp[l]){
            k=i;
            break;
        }
    }
    dis(l+1,k-1);
    dis(k,r);
    cout<<" "<<dp[l];
    return ;
}

void solve(){
    int n;
    cin>>n;
    rep(i,1,n) cin>>dp[i];
//    bool flag=0;
    if(dp[1]>dp[2]){
        bool flag=judge(1,n);
        if(flag){
           cout<<"YES"<<endl;
            post(1,n);
            cout<<endl;
        }
        else cout<<"NO"<<endl;
    }
    else {
        bool flag=check(1,n);
        if(flag){
           cout<<"YES"<<endl;
            dis(1,n);
            cout<<endl;
        }
        else cout<<"NO"<<endl;
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
	return 0;
}