pat甲级——1043 Is It a Binary Search Tree (25分)

题意：给出一组数，判断是否为二叉搜索树的前序遍历，或者为二叉搜索树的镜面前序遍历

题解：

法一：直接建树，然后前序遍历，并在遍历过程判断是否与题目给出的数相等

          如果不相等，可以发现镜面的前序遍历为根右左，按照这个顺序再遍历一次即可

          解法麻烦一点但好理解

        

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e3+5;
int p[maxn],a[maxn];
int n;
int fg;
int m;
struct tree{
	int da;
	tree *le,*ri;
};
void insert(int da,tree* &rt){
	if(rt==NULL){
		rt=new tree();
		rt->da=da;
		rt->le=NULL;
		rt->ri=NULL;
		return ;
	}
	if(da>=rt->da) insert(da,rt->ri);
	if(da<rt->da)  insert(da,rt->le);
}
void pre(tree* rt){
	static int k=0;
	if(rt==NULL) return ;
	if(rt->da!=a[++k]) fg=0;
	pre(rt->le);
	pre(rt->ri);
	p[++m]=rt->da;
}
void post(tree *rt){
	if(rt==NULL) return ;
	static int k=0;
//	cout<<rt->da<<" ";
	if(rt->da!=a[++k]) fg=0;
	post(rt->ri);
	post(rt->le);
	p[++m]=rt->da;
}
int main(){
	int n;
	cin>>n;
	tree *root=NULL;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		insert(a[i],root);
	}
	fg=1;
	pre(root);
	if(fg){
		cout<<"YES\n";
    	for(int i=1;i<m;i++)
    	 cout<<p[i]<<" ";
    	cout<<p[m]<<endl; 
	}
	else if(!fg){
		fg=1;
		m=0;
		post(root);
		if(fg){
		  cout<<"YES\n";
    	  for(int i=1;i<m;i++)
    	   cout<<p[i]<<" ";
          cout<<p[m]<<endl; 
		}
		else cout<<"NO\n";
	} 
	return 0;	
}

法二：先假设给出的为搜索树的前序遍历，利用前序遍历得到后续遍历，如果后续遍历结果小于n，则假设为镜面前序遍历

得到后续遍历，如果也小于n,则输出NO

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e3+5;
int a[maxn];
bool is_mirror=0;
vector<int> v;
void getpost(int l,int r){
	int	i=l+1,j=r;
	if(!is_mirror){
		while(a[i]<a[l]&&i<=r) i++;
		while(a[j]>=a[l]&&j>l) j--;
	}
	else{
		while(a[i]>=a[l]&&i<=r) i++;
		while(a[j]<a[l]&&j>l) j--;
	}
	if(i-j!=1) return ;
	getpost(l+1,j);
	getpost(i,r);
	v.push_back(a[l]);
}
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	 cin>>a[i];
    getpost(1,n);
    if(v.size()!=n){
    	is_mirror=1;
    	v.clear();
    	getpost(1,n);
	}
	if(v.size()==n){ 
		cout<<"YES"<<endl;
		for(int i=0;i<v.size();i++){
			if(i) cout<<" ";
			cout<<v[i]; 
		}
	}
	else cout<<"NO"<<endl;
	return 0;
}