题目链接: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;
}