对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数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
思路:
首先通过先序序列生成二叉搜索树/镜像二叉搜索树,对该树进行判断是否是正常的;
先序序列的特点:首点即为根节点,从该点往后扫描序列,第一个比它大或者等于的右边部分就是右子树,在该点左边部分就是左子树;递归生成搜索树;
题目中提到的镜像二叉搜索树同理,只不过条件相反,第一个比它小的右边部分就是右子树..
判断二叉搜索树是否正常就是在该树的根节点右侧寻找是否有比它小的节点存在,如果存在那么就是不正常的。这里用全局变量flag进行标记。
镜像二叉搜索树的判断同理;条件相反,找的是是否有大于等于它的节点。用全局变量flag_m进行标记
#include<stdio.h>
#include<stdlib.h>
#define MAXSIZE 1010
//通过先序序列生成二叉搜索树
//根节点必定为序列第一个
int flag=0;//正常二叉搜索树是否正常
int flag_m=0;//镜像二叉搜索树是否正常
int tip=0;//在最后的后序遍历中输出序列时去掉最后一个空格
typedef struct tree
{
int data;
struct tree *left,*right;
}*Tree;
Tree find(int pre[],int n)//查找正向的
{
if(n==0)return NULL;
Tree t=(Tree)malloc(sizeof(Tree));
t->data=pre[0];
int index=1,j;
while(index<n)
{
if(pre[index]>=pre[0])break;
index++;
}
for(j=index;j<n;j++)
{
if(pre[j]<t->data)
{
flag=1;
return NULL;
}
}
t->left=find(pre+1,index-1);
t->right=find(pre+index,n-index);
return t;
}
Tree findm(int pre[],int n)//查找镜像
{
if(n==0)return NULL;
Tree t=(Tree)malloc(sizeof(Tree));
t->data=pre[0];
int index=1,j;
while(index<n)
{
if(pre[index]<pre[0])break;
index++;
}
for(j=index;j<n;j++)
{
if(pre[j]>=t->data)
{
flag_m=1;
return NULL;
}
}
t->left=findm(pre+1,index-1);
t->right=findm(pre+index,n-index);
return t;
}
void printt(Tree t)
{
if(t)
{
printt(t->left);
printt(t->right);
if(tip==0)
{
tip=1;
printf("%d",t->data);
}
else printf(" %d",t->data);
}
}
int main()
{
int i,n,pre[MAXSIZE];
Tree t,tm;
scanf("%d",&n);
for(i=0;i<n;i++)scanf("%d",&pre[i]);
t=find(pre,n);
tm=findm(pre,n);
if(!flag&&t)
{
printf("YES\n");
printt(t);
}
else if(!flag_m&&tm)
{
printf("YES\n");
printt(tm);
}
else printf("NO\n");//两个都不正常
return 0;
}