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

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

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

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

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

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

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

#include <iostream>
#include <vector>
using namespace std;
int n;

//前序序列
int pre[1010];

//后序序列
vector<int> a;

//l、r区间左右端点 
//f标记镜像或非镜像
//每次递归将区间划分为三块 root:l left:[l + 1, x - 1] right:[x, r];
bool dfs(int l, int r, bool f){
    if (l > r) return true;
    int res = true;
    int x;
    if(f) {
        //找到划分处x
        for(x = l + 1; x <= r; x++) if(pre[x] < pre[l]) break;
        //判断是否合法
        if(x <= r) for(int j = x; j <= r; j++) if(pre[j] >= pre[l]) return false;
    } else {
        for(x = l + 1; x <= r; x++) if(pre[x] >= pre[l]) break;
        if(x <= r) for(int j = x; j <= r; j++) if(pre[j] < pre[l]) return false;
    }
    if (x > l + 1){
        res = res && dfs(l + 1, x - 1, f);
    }
    if(x <= r){
        res = res && dfs(x, r, f);
    }
    //对root的操作放最后 可以得到后序遍历的结果
    a.push_back(pre[l]);
    return res;
}

int main() {
    cin >> n;
    for(int i = 1; i <= n;i++) {
        cin >> pre[i];
    }
    bool st = dfs(1, n, true);
    if(!st) {
        //清空数组
        a.clear();
        st = dfs(1, n, false);
    }
    if(st){
        cout << "YES" << endl;
        for(int i = 0; i < n; i++){
            if(i == 0) cout << a[i];
            else cout << " " << a[i];
        }
        cout << endl;
    } else{
        cout << "NO" << endl;
    }
    return 0;
}