L2-004 这是二叉搜索树吗？（c++）

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

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

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

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

输入格式：

输入的第一行给出正整数 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

分析：

因为前序遍历是中->左->右，而二叉搜索树又是左<中<=右，所以可以根据前序遍历的结果数组a找到第一个大于等于a[0]的元素，如果该元素后面还有小于a[0]的元素，则该树不是二叉搜索树，否则继续判断它的左右子树。

c++代码

#include "iostream"
using namespace std;

typedef struct Node* tree;

struct Node {
	int data;
	tree l;
	tree r;
};

int check1 = 0, check2 = 0;

// 检查是不是二叉搜索树
tree istree1(int a[], int len) {
	int i; // 用i来储存第一个大于等于a[0]的元素的下标
	if (!len)return NULL;
	tree t = (tree)malloc(sizeof(struct Node));
	t->data = a[0];
	for (i = 1;i < len;i++) { // 注意i从1开始
		if (a[i] >= t->data)break;
	}
	// 如果i之后还有比a[0]小的元素，则不是二叉搜索树
	for (int j = i;j < len;j++) {
		if (a[j] < t->data) {
			check1 = 1;
			return NULL;
		}
	}
	t->l = istree1(a + 1, i - 1); // a[1]到a[i-1]为左子树
	t->r = istree1(a + i, len - i); // a[i]到a[len-1]为右子树
	return t;
}

// 依样画葫芦，判断是不是二叉搜索树的“镜像”
tree istree2(int a[], int len) {
	int i; // 用i来储存第一个小于a[0]的元素的下标
	if (!len)return NULL;
	tree t = (tree)malloc(sizeof(struct Node));
	t->data = a[0];
	for (i = 1;i < len;i++) { // 注意i从1开始
		if (a[i] < t->data)break;
	}
	// 如果i之后还有大于等于a[0]的元素，则不是二叉搜索树
	for (int j = i;j < len;j++) {
		if (a[j] >= t->data) {
			check2 = 1;
			return NULL;
		}
	}
	t->l = istree2(a + 1, i - 1); // a[1]到a[i-1]为左子树
	t->r = istree2(a + i, len - i); // a[i]到a[len-1]为右子树
	return t;
}

// 输出后序遍历的结果
int flag = 1;
void print(tree t) {
	if (t) {
		print(t->l); // 输出左子树
		print(t->r); // 输出右子树
		if (flag)flag = 0;
		else cout << " ";
		cout << t->data; // 输出当前结点的键值
	}
}

int main() {
	int n;
	int a[1005];
	cin >> n;
	for (int i = 0;i < n;i++)cin >> a[i];
	tree t1 = istree1(a, n);
	tree t2 = istree2(a, n);
	if (t1 && !check1) {
		cout << "YES" << endl;
		print(t1);
	}
	else if (t2 && !check2) {
		cout << "YES" << endl;
		print(t2);
	}
	else {
		cout << "NO" << endl;
	}
}