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

题目链接：https://www.patest.cn/contests/gplt/L2-004

题目：

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

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

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

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

输入格式：

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

题意：给出一个序列判断是否是某个二叉搜索树或者其镜像的前序遍历序列。

题解：二叉搜索树的左子树都比根小，右子树都比根大，判断从第二个元素开始其后的元素能否分成两个连续的序列，一个都比他小，一个都比他大。自己写的总是部分错误，可能是边界问题吧。参考：http://blog.csdn.net/liangzhaoyang1/article/details/51388494

输出后序序列，递归反转。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX=1000+10;
int order[MAX];
int N;
int cnt;
bool judge1(int l,int r)
{
	if(l>=r) return true;
	int root=order[l];
	int i;
	int fl=0,fr=0;
	for(i=l;i<=r;i++)
	{
		fl=1;
		for(int j=i;j>l;j--)
		{
			if(order[j]>=root) 
			{
				fl=0;
				break;
			}
		}
		fr=1;
		for(int j=i+1;j<=r;j++)
		{
			if(order[j]<root)
			{
				fr=0;
				break;
			}
		}
		if(fl&&fr)
			break;
	}
	if(fl&&fr)
		return judge1(l+1,i)&&judge1(i+1,r);
	else 
		return false;
}
bool judge2(int l,int r)
{
	if(l>=r) return true;
	int root=order[l];
	int i;
	int fl=0,fr=0;
	for(i=l;i<=r;i++)
	{
		fl=1;
		for(int j=i;j>l;j--)
		{
			if(order[j]<root) 
			{
				fl=0;
				break;
			}
		}
		fr=1;
		for(int j=i+1;j<=r;j++)
		{
			if(order[j]>=root)
			{
				fr=0;
				break;
			}
		}
		if(fl&&fr)
			break;
	}
	if(fl&&fr)
		return judge2(l+1,i)&&judge2(i+1,r);
	else 
		return false;
}
void getpost1(int l,int r)
{
	if(r<l) return ; 
	int root=order[l];
	if(l==r) 
	{
		printf(++cnt==1?"%d":" %d",root);
		return;
	}
	int i;
	int fl=0,fr=0;
	for(i=l;i<=r;i++)
	{
		fl=1;
		for(int j=i;j>l;j--)
		{
			if(order[j]>=root) 
			{
				fl=0;
				break;
			}
		}
		fr=1;
		for(int j=i+1;j<=r;j++)
		{
			if(order[j]<root)
			{
				fr=0;
				break;
			}
		}
		if(fl&&fr)
			break;
	}
	//printf("%d %d %d %d\n",l+1,mid-1,mid,r);
	getpost1(l+1,i);
	getpost1(i+1,r);
	printf(++cnt==1?"%d":" %d",root);
}
void getpost2(int l,int r)
{
	if(r<l) return ; 
	int root=order[l];
	if(l==r) 
	{
		printf(++cnt==1?"%d":" %d",root);
		return ;
	}
	int i;
	int fl=0,fr=0;
	for(i=l;i<=r;i++)
	{
		fl=1;
		for(int j=i;j>l;j--)
		{
			if(order[j]<root) 
			{
				fl=0;
				break;
			}
		}
		fr=1;
		for(int j=i+1;j<=r;j++)
		{
			if(order[j]>=root)
			{
				fr=0;
				break;
			}
		}
		if(fl&&fr)
			break;
	}
	//printf("%d %d %d %d\n",l+1,mid-1,mid,r);
	getpost2(l+1,i);
	getpost2(i+1,r);
	printf(++cnt==1?"%d":" %d",root);
}
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	while(scanf("%d",&N)!=EOF)
	{
		for(int i=0;i<N;++i)
		{
			scanf("%d",&order[i]);
		}
		if(judge1(0,N-1))
		{
			printf("YES\n");
			cnt=0;
			getpost1(0,N-1);
			putchar(10);
		}
		else if(judge2(0,N-1))
		{
			printf("YES\n");
			cnt=0;
			getpost2(0,N-1);
			putchar(10);
		}
		else printf("NO\n");
	}
	return 0;
}