PAT (Advanced Level) 1043. Is It a Binary Search Tree (25)

简单题。构造出二叉搜索树，然后check一下。

#include<stdio.h>
#include<algorithm>
using namespace std;

const int maxn=+;
struct Node
{
    int left;
    int right;
    int val;
} s[maxn];

int n;
int a[maxn];
int ans[maxn],tot;
int h[maxn];
int k;

void dfs(int x)
{
    ans[k++]=s[x].val;
    if(s[x].left!=-) dfs(s[x].left);
    if(s[x].right!=-) dfs(s[x].right);
}

void houxu(int x)
{
    if(s[x].left!=-) houxu(s[x].left);
    if(s[x].right!=-) houxu(s[x].right);
    h[k++]=s[x].val;
}

bool check()
{
    for(int i=;i<n;i++) if(ans[i]!=a[i]) return ;
    return ;
}

int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=; i<n; i++) scanf("%d",&a[i]);
        for(int i=; i<=n; i++) s[i].left=s[i].right=-;
        int id=;
        s[id++].val=a[];
        for(int i=; i<n; i++)
        {
            int now=;
            while()
            {
                if(a[i]<s[now].val)
                {
                    if(s[now].left!=-) now=s[now].left;
                    else
                    {
                        s[now].left=id;
                        s[id++].val=a[i];
                        break;
                    }
                }
                else
                {
                    if(s[now].right!=-) now=s[now].right;
                    else
                    {
                        s[now].right=id;
                        s[id++].val=a[i];
                        break;
                    }
                }
            }
        }

       // for(int i=0;i<id;i++) printf("%d %d %d\n",s[i].val,s[i].left,s[i].right);

        k=;
        dfs();
        if(check()==)
        {
            printf("YES\n");
            k=; houxu();
            for(int i=;i<n;i++)
            {
                printf("%d",h[i]);
                if(i<n-) printf(" ");
                else printf("\n");
            }
            continue;
        }

        k=;
        for(int i=;i<n;i++) swap(s[i].left,s[i].right);
        dfs();
        if(check()==)
        {
            printf("YES\n");
            k=; houxu();
            for(int i=;i<n;i++)
            {
                printf("%d",h[i]);
                if(i<n-) printf(" ");
                else printf("\n");
            }
            continue;
        }

        printf("NO\n");

    }
    return ;
}