哎真是一天不如一天了,本来早就会的题,现在做却不会了
对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数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
#include <bits/stdc++.h>
using namespace std;
#define N 1005
int flag1=0,flag2=0;
typedef struct node* List;
struct node{
List left;
List right;
int data;
};
int a[N]={0};
List find (int *a,int n)
{
int i,j;
if(!n) return NULL;
List temp=(List)malloc(sizeof(struct node));
temp->data=a[0];
for(i=1;i<n;i++){//往后找第一个比根大或者等于的数
if(temp->data<=a[i])
break;
}
for(j=i;j<n;j++){
if(a[j]<temp->data){//找如果右子树中如果有小于他的,就标记
flag1=1;
return NULL;
}
}
temp->left=find(a+1,i-1);
temp->right=find(a+i,n-i);
return temp;
}
List findmicro (int *a,int n)
{
int i,j;
if(!n) return NULL;
List temp=(List)malloc(sizeof(struct node));
temp->data=a[0];
for ( i = 1 ; i < n ; i++ ) {
if(a[i]<temp->data){
break;
}
}
for( j = i ; j < n ; j++ ){
if(a[j]>=temp->data){
flag2=1;
return NULL;
}
}
temp->left=findmicro(a+1,i-1);
temp->right=findmicro(a+i,n-i);
return temp;
}
int flag3=0;
void print(List a)
{
if(a->left) print(a->left);
if(a->right) print(a->right);
if(!flag3) {
cout<<a->data;
flag3=1;
}
else cout<<' '<<a->data;
}
int main()
{
int n;
cin>>n;
int a[n];
for(int i=0;i<n;i++){
cin>>a[i];
}
List t=find(a,n);
List T=findmicro(a,n);
if(t&&!flag1){
cout<<"YES"<<endl;
print(t);
}
else if(T&&!flag2) {
cout << "YES" << endl;
print(T);
}
else
cout<<"NO"<<endl;
return 0;
}