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;
}