1043 Is It a Binary Search Tree (25 分)

PAT目录

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=1010;
int n,cnt;//后序遍历用cnt
int preorder[N],inorder[N],postorder[N];
bool built(int il,int ir,int pl,int pr,int type){
    if(il>ir) return true;//若为空则一定可以建立成功
    int root=preorder[pl];
    int k;
    if(type==0){//原树
        for(k=il;k<=ir;k++)
            if(inorder[k]==root) break;
        if(k>ir) return false;
    }else{
        for(k=ir;k>=il;k--)
            if(inorder[k]==root) break;
        if(k<il) return false;
    }
    bool res=true;
    if(!built(il,k-1,pl+1,pl+1+k-1-il,type))
        res=false;
    if(!built(k+1,ir,pl+1+k-1-il+1,pr,type))
        res=false;
    postorder[cnt++]=root;
    return res;
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>preorder[i];
        inorder[i]=preorder[i];
    }
    sort(inorder,inorder+n);//排序后为中序遍历，BST中序遍历是有序的
    if(built(0,n-1,0,n-1,0)){//原树是一个BST
        puts("YES");
        cout<<postorder[0];
        for(int i=1;i<n;i++) cout<<' '<<postorder[i];
        cout<<endl;
    }
    else{
        reverse(inorder,inorder+n);
        cnt=0;
        if(built(0,n-1,0,n-1,1)){
            puts("YES");
            cout<<postorder[0];
            for(int i=1;i<n;i++) cout<<' '<<postorder[i];
            cout<<endl;
        }
        else{
            puts("NO");
        }
    }
}