7-68 搜索树判断 (25 分)
对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数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
前序遍历是DLR,然后又因为二叉搜索树的性质,从左到右第一个大于等于pre[0]的数就是右子树,如果右子树里面有小于pre[0]的数,那么就不是二叉搜索树,如果没有,那就构建一棵树
#include<iostream>
using namespace std;
typedef struct nade
{
int data;
struct nade *lchild;
struct nade *rchild;
}DATE,*tree;
int flag=0;//结尾无空格
int flag1=0;//判断是否为二叉搜索树
int flag2=0;//判断是否为镜像二叉搜索树
tree find(int pre[],int n)//判断是否为搜索树,并且创建一棵树
{
if(!n) return NULL;
tree root=(tree)malloc(sizeof(DATE));
root->data =*pre;
int i=1,j;
for(;i<n;i++)//先找到右子树的位置
{
if(pre[i]>=root->data )
{
break;
}
}
for(j=i;j<n;j++)
{
if(pre[j]<root->data )
{
flag1=1;//不是搜索树,记录下来返回NULL
return NULL;
}
}
//建立左右子树
root->lchild =find(pre+1,i-1);
root->rchild =find(pre+i,n-i);
return root;
}
tree mirror_find(int pre[],int n)//镜像搜索树跟搜索树一样
{
if(!n) return NULL;
tree root=(tree)malloc(sizeof(DATE));
root->data =*pre;
int i=1,j;
for(;i<n;i++)
{
if(pre[i]<root->data )
{
break;
}
}
for(j=i;j<n;j++)
{
if(pre[j]>=root->data )
{
flag2=1;
break;
}
}
root->lchild =mirror_find(pre+1,i-1);
root->rchild =mirror_find(pre+i,n-i);
return root;
}
void print(tree root)
{
if(root==NULL) return ;
print(root->lchild );
print(root->rchild );
if(!flag) flag=1;
else cout<<" ";
cout<<root->data ;
}
int main()
{
int n;
cin>>n;
int pre[1010]={0};
for(int i=0;i<n;i++)
{
cin>>pre[i];
}
tree ta=find(pre,n);
tree tb=mirror_find(pre,n);
if(ta&&!flag1)
{
cout<<"YES"<<endl;
print(ta);
}
else if(tb&&!flag2)
{
cout<<"YES"<<endl;
print(tb);
}
else cout<<"NO"<<endl;
return 0;
}