7-28 查找树判断

给出一棵树的前序遍历←→求后序，中序遍历。

代码：

#include<iostream>
using namespace std;
const int N=1010;
int n,idx;
int a[N],ans[N];
void dfs1(int l,int r){
	if(l>r){ return ;}
	int i,j;
	for(i=l+1;i<=r;i++)
	   if(a[i]>=a[l]) break;
	for(j=i;j<=r;j++)
	   if(a[j]<a[l]) return;
	dfs1(l+1,i-1);
	dfs1(i,r);
	ans[++idx]=a[l];
//	dfs1(l+1,i-1);   ---中序 
//	ans[++idx]=a[l];
//	dfs1(i,r);
}
void dfs2(int l,int r){
	if(l>r){ return ;}
	int i,j;
	for(i=l+1;i<=r;i++)
	   if(a[i]<a[l]) break;
	for(j=i;j<=r;j++)
	   if(a[j]>=a[l]) return;
	dfs2(l+1,i-1);
	dfs2(i,r);
	ans[++idx]=a[l];
}
void prit(){
	printf("YES\n");
    for(int i=1;i<n;i++)
    	printf("%d ",ans[i]);
    printf("%d",ans[n]);
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++)
	   scanf("%d",&a[i]);
    dfs1(0,n-1);
    if(idx==n){prit(); return 0;}
    idx=0;
    dfs2(0,n-1);
    if(idx==n){prit(); return 0;}
    printf("NO");
    return 0;
}