树的遍历——pat1043

http://pat.zju.edu.cn/contests/pat-a-practise/1043

给予N个数字组成二叉搜索树，判断这个数列是否由先序遍历得出或是镜像先序遍历得出，若是则输出相应的后续遍历或是镜像后续遍历

分析：

镜像先序遍历其实就是 先访问祖先，再访问右子树，再左子树

镜像后续遍历就是先访问右子树，再左子树，在祖先

#include<stdio.h>

struct tree{
    int v;
    tree *left,*right;
};

int shu[];
int pre[];
int post[];
int step;

void bulid(tree* head,int v){
    if(v >= head->v){
        if(head->right==NULL){
            head->right=new tree;
            head=head->right;
            head->v=v;
            head->left=NULL;
            head->right=NULL;
        }else{
            bulid(head->right,v);
        }
    }else{
        if(head->left==NULL){
            head->left=new tree;
            head=head->left;
            head->v=v;
            head->left=NULL;
            head->right=NULL;
        }else{
            bulid(head->left,v);
        }
    }
}

void preTree(tree* head,int mirror){
    pre[step]=head->v;
    step++;
    if(mirror==){
        if(head->left!=NULL)preTree(head->left,mirror);
        if(head->right!=NULL)preTree(head->right,mirror);
    }else{
        if(head->right!=NULL)preTree(head->right,mirror);
        if(head->left!=NULL)preTree(head->left,mirror);
    }
}

void postTree(tree* head,int mirror){
    if(mirror==){
        if(head->left!=NULL)postTree(head->left,mirror);
        if(head->right!=NULL)postTree(head->right,mirror);
    }else{
        if(head->right!=NULL)postTree(head->right,mirror);
        if(head->left!=NULL)postTree(head->left,mirror);
    }
    post[step]=head->v;
    step++;
}

int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        int i,mirror=-;

        tree* rhead;
        tree* head=new tree;
        rhead=head;

        for(i=;i<=n;i++){
            scanf("%d",&shu[i]);
        }
        if(n==){
            printf("YES\n");
            printf("%d\n",shu[]);
            continue;
        }

        head->v=shu[];
        head->left=NULL;
        head->right=NULL;
        for(i=;i<=n;i++){
            bulid(rhead,shu[i]);
        }

        step=;
        preTree(rhead,);//普通的先序遍历
        for(i=;i<=n;i++){
            if(shu[i]!=pre[i])break;
        }
        if(i==n+)mirror=;

        step=;
        preTree(rhead,);//镜像先序遍历
        for(i=;i<=n;i++){
            if(shu[i]!=pre[i])break;
        }
        if(i==n+)mirror=;

        if(mirror==-){
            printf("NO\n");
            continue;
        }

        printf("YES\n");
        step=;
        postTree(rhead,mirror);
        int ok=;
        for(i=;i<=n;i++){
            if(ok==)ok=;
            else printf(" ");
            printf("%d",post[i]);
        }
        printf("\n");
    }

    return ;
}