3 搜索树判断

3 搜索树判断

分数 25
作者 DS课程组
单位 浙江大学

对于二叉搜索树，我们规定任一结点的左子树仅包含严格小于该结点的键值，而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树，得到的树叫做镜像二叉搜索树。

现在我们给出一个整数键值序列，请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，如果是，则输出对应二叉树的后序遍历序列。

输入格式:

输入的第一行包含一个正整数N（≤1000），第二行包含N个整数，为给出的整数键值序列，数字间以空格分隔。

输出格式:

输出的第一行首先给出判断结果，如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，则输出YES，否侧输出NO。如果判断结果是YES，下一行输出对应二叉树的后序遍历序列。数字间以空格分隔，但行尾不能有多余的空格。

输入样例1:

7
8 6 5 7 10 8 11

输出样例1:

YES
5 7 6 8 11 10 8

输入样例2:

7
8 6 8 5 10 9 11

输出样例2:

NO

代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
C++ (g++)

思路：

①判断是不是二叉搜索树/镜像二叉搜索树
给出了先序遍历，首元素为根
二叉搜索树：
左子树上的点都比根小，右子树上的点都比数大。
找到第一个比根大的数 它以及它后面的都是右子树，都要比根大，否则不是二叉搜索树
镜像二叉搜索树：
同理，镜像二叉搜索树的意思二叉搜索树倒个。
就是找到第一个比根小的数 它以及他后面的都是右子树，都要比根小，否则不是镜像二叉搜索树。
②输出
如果是的话输出后序遍历序列。左右根。注意结尾不能有多余空格。

AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef struct node *tree;
struct node {
	int data;
	tree left;
	tree right;
};
int n;
int pre[1005], flag1 = 1, flag2 = 1;
int cnt = 0;
void Print(tree t) {
	if (t) {
		Print(t->left);
		Print(t->right);
		if (cnt++)
			cout << " ";
		cout << t->data;
	}
}
tree setTree(int pre[], int n) {
	if (n <= 0)
		return NULL;
	tree t = new node;
	t->data = pre[0];
	int i, j;
	for (i = 1; i < n; i++) {
		if (pre[i] >= t->data)
			break;
	}
	for (j = i; j < n; j++) {
		if (pre[j] < t->data) {
			flag1 = 0;
			return NULL;//不是二叉搜索树
		}
	}
	t->left = setTree(pre + 1, i - 1);
	t->right = setTree(pre + i, n - i);
	return t;
}
tree setMirroTree(int pre[], int n) {
	if (n <= 0)
		return NULL;
	tree t = new node;
	t->data = pre[0];
	int i, j;
	for (i = 1; i < n; i++) {
		if (pre[i] < t->data)
			break;
	}
	for (j = i; j < n; j++) {
		if (pre[j] >= t->data) {
			flag2 = 0;
			return NULL;//不是镜像二叉搜索树
		}
	}
	t->left = setMirroTree(pre + 1, i - 1);
	t->right = setMirroTree(pre + i, n - i);
	return t;
}
int main() {
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> pre[i];
	tree t1, t2;
	t1 = setTree(pre, n);
	t2 = setMirroTree(pre, n);
	if (t1 && flag1) {
		cout << "YES" << endl;
		Print(t1);
	}
	else if (t2 && flag2) {
		cout << "YES" << endl;
		Print(t2);
	}
	else
		cout << "NO" << endl;
	return 0;
}