L2-004. 这是二叉搜索树吗？ （二叉搜索树的性质）

L2-004. 这是二叉搜索树吗？

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

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

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

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

输入格式：

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

//根据前序遍历的性质每次可以把根节点分离出来；根据二叉搜索树的性质每次可以把左右子树分开；这样在判断的时候只判断左边的是不是都小于根，右边的是不是都大于根就行了。至于后续序列，在递归左右之后，用post数组保存。
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
bool flag = true;
int post_index = 0;
vector<int> pre(1001);
vector<int> post(1001);

bool check(int l, int r)
{
	if (l > r)
		return true;
	int father = pre[l]; //父节点
	int i = l + 1, j; //i为左子树的起点
	if (flag) 
	{
		while (i <= r && pre[i] < father)
			i++;
		j = i; // j为右子树的起点
		while (j <= r)
		{
			if (pre[j] < father) // 题目确定右边的子树能等于父节点，其实定义里是不行的
				return false;
			j++;
		}
	}
	else
	{
		while (i <= r && pre[i] >= father)
			i++;
		j = i;
		while (j <= r)
		{
			if (pre[j] >= father)
				return false;
			j++;
		}
	}
	if (!check(l + 1, i - 1)) // 检查左子树
		return false;
	if (!check(i, r)) // 检查右子树
		return false;
	post[post_index++] = father;
	return true;
}


int main()
{
	//freopen("data.txt", "r", stdin);
	int N;
	cin >> N;
	for (int i = 0; i < N; i++)
		scanf("%d", &pre[i]);

	if (N == 1) {
		cout << "YES" << endl << pre[0] << endl;
		return 0;
	}

	if (pre[0] <= pre[1])
		flag = !flag;

	if (check(0, N - 1))
	{
		cout << "YES" << endl;
		for (int i = 0; i < post_index - 1; i++)
			cout << post[i] << " ";
		cout << post[post_index - 1] << endl;
	}
	else
		cout << "NO" << endl;
	return 0;
}