【题意】给定一个序列,判断是否为二叉搜索树的前序遍历序列 或 其镜像的前序遍历;
【分析】
- 建树及遍历模板题(模板戳这) 该题要两个build函数分别建BST和BST镜像;
- 注意插入的时候,当前节点的值val与x的比较不要写错,x是要插入的值,是要与val进行比较的。一开始我这里就出现了些小问题大于小于写反了;
- 求pre和post数组时,一开始是将cnt做参数传进,每次++然后对数组进行赋值,但这样会有问题,就写两个全局变量cnt1和cnt2;
- 注意每次调用getPre()之前cnt1要初始化!还有T1、T2要初始化为NULL!!
- ///在想会不会BST和BST的镜像都符合题意,但是想想感觉不大可能。嗯~
#include<bits/stdc++.h>
using namespace std;
const int maxn=1005;
int pre[maxn],post[maxn],a[maxn],cnt1=0,cnt2=0;
typedef struct node{
int val;
struct node *left;
struct node *right;
}node,*tree;
void insert1(tree &T,int x)
{
if(T==NULL)
{
T=new node;
T->val=x;
T->left=T->right=NULL;
}
else if(x<T->val)
insert1(T->left,x);
else if(x>=T->val)
insert1(T->right,x);
}
void insert2(tree &T,int x)
{
if(T==NULL)
{
T=new node;
T->val=x;
T->left=T->right=NULL;
}
else if(x>=T->val)
insert2(T->left,x);
else if(x<T->val)
insert2(T->right,x);
}
void postT(tree T)
{
if(T)
{
postT(T->left);
postT(T->right);
post[cnt2++]=T->val;
}
}
void preT(tree T)
{
if(T)
{
pre[cnt1++]=T->val;
preT(T->left);
preT(T->right);
}
}
int main()
{
int n,flag1=1,flag2=1;
tree t1=NULL,t2=NULL;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
insert1(t1,a[i]);
insert2(t2,a[i]);
}
preT(t1);
for(int i=0;i<n;i++)
{
if(a[i]!=pre[i])
{
flag1=0;
break;
}
}
if(flag1)postT(t1);
else{
flag2=1;
cnt1=cnt2=0;
preT(t2);
for(int i=0;i<n;i++)
{
if(a[i]!=pre[i])
{
flag2=0;break;
}
}
if(flag2)postT(t2);
}
if(flag1 || flag2)
{
cout<<"YES\n"<<post[0];
for(int i=1;i<n;i++)
cout<<" "<<post[i];
cout<<endl;
}
else cout<<"NO"<<endl;
return 0;
}