1043. Is It a Binary Search Tree (25)

    直接对序列进行判断，第一个肯定是根结点（值为 val），然后如果第一个小于第二个元素则先假设该输入序列为一BST镜像树的preorder，若假设为BST则

从序列的第二个元素开始，查找第一个 >= val 的元素，判断其后所有的元素是否也都 >= val
是，则递归查找左子树，右子树，然后将val存储（模拟后序遍历）
否，则直接return

    若假设为BST的镜像树，则

从序列的第二个元素开始，查找第一个 < val 的元素，判断其后所有的元素是否也都 < val
是，则递归查找左子树，右子树，然后存储 val
否，则直接return

    最后直接判断存储元素的个数是否等于输入序列个数，是则为一个BST或者一个BST的镜像树，否则不是。

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <functional>

using namespace std;

vector<int> result;
int rootValue;
function<int(int)> fn = [&](int val){ return val >= rootValue; };

template<typename Iter>
void tranversal(Iter preBegin, Iter preEnd){
if(preBegin == preEnd) return;

rootValue = *preBegin;
auto leftEnd = find_if(next(preBegin), preEnd, fn);

if(!all_of(leftEnd, preEnd, fn)) return;

tranversal(next(preBegin), leftEnd);
tranversal(leftEnd, preEnd);
result.push_back(*preBegin);
}

int main(){
int n;
scanf("%d", &n);

vector<int> seq(n);
for(int i = 0; i < n; ++i){
scanf("%d", &seq[i]);
}

if(n > 1 && seq[0] <= seq[1]){
fn = [&](int val){ return val < rootValue; };
}

tranversal(begin(seq), end(seq));
if((int)result.size() != n) printf("NO\n");
else{
printf("YES\n");
for(size_t i = 0; i < result.size(); ++i){
if(i) printf(" ");
printf("%d", result[i]);
}
}

return 0;
}