团体程序设计天梯赛 L3-016. 二叉搜索树的结构

 #include <cstdio>
 #include <cstdlib>
 #include <string.h>
 #include <math.h>
 #include <stdbool.h>
 #include <algorithm>
 using namespace std;
 #define maxn 500+100

 //离散化+结构体数组存储树(数据可能很大，不可能用tree[x]记录值为x的结点，而用链表存储树，在这一题判断兄弟结点等不太方便)
 //输入的数据点也许并不在树上
 //兄弟结点不能相同

 struct node
 {
     long left,right,fa,depth;
 }tree[maxn];

 long n,num[maxn],z[maxn];

 bool cmp(long a,long b)
 {
     return a<b;
 }

 //better:binary search
 int findnum(long value)
 {
     long i;
     for (i=;i<=n;i++)
         if (value==z[i])
             return i;
     return n+;
 }

 int add(long pos,long value,long depth)
 {
     if (value < pos)
     {
         if (tree[pos].left==)
         {
             tree[pos].left=value;
             tree[value].fa=pos;
             tree[value].depth=depth;
         }
         else
             add(tree[pos].left,value,depth+);
     }
     else
     {
         if (tree[pos].right==)
         {
             tree[pos].right=value;
             tree[value].fa=pos;
             tree[value].depth=depth;
         }
         else
             add(tree[pos].right,value,depth+);
     }
 }

 int main()
 {
     long i,j,x,y,p,q,m,root;
     char a[],b[],c[],d[],e[],f[];
     scanf("%ld",&n);
     for (i=;i<=n;i++)
     {
         scanf("%ld",&z[i]);
         num[i]=z[i];
     }
     for (i=;i<=n;i++)
     {
         tree[i].left=;
         tree[i].right=;
         tree[i].fa=; //不能像并查集一样设置为i
     }
     sort(z+,z+n+,cmp); //[x,y)

     root=findnum(num[]);
     tree[root].depth=;
     for (i=;i<=n;i++)
     {
         j=findnum(num[i]);
         add(root,j,);
     }

     scanf("%ld",&m);
     while (m)
     {
         m--;
         scanf("%ld",&x);
         p=findnum(x);
         scanf("%s",a);
         if (strcmp(a,"is")==)
         {
             scanf("%s%s",b,c);
             if (strcmp(c,"root")==)
             {
 //                if (p!=n+1 && tree[p].fa==0)
                 if (x==num[])
                     printf("Yes\n");
                 else
                     printf("No\n");
             }
             else if (strcmp(c,"parent")==)
             {
                 scanf("%s",d);
                 scanf("%ld",&y);
                 q=findnum(y);
                 if (q!=n+ && tree[q].fa==p)
                     printf("Yes\n");
                 else
                     printf("No\n");
             }
             else if (strcmp(c,"left")==)
             {
                 scanf("%s%s",d,e);
                 scanf("%ld",&y);
                 q=findnum(y);
                 if (q!=n+ && tree[q].left==p)
                     printf("Yes\n");
                 else
                     printf("No\n");
             }
             else
             {
                 scanf("%s%s",d,e);
                 scanf("%ld",&y);
                 q=findnum(y);
                 if (q!=n+ && tree[q].right==p)
                     printf("Yes\n");
                 else
                     printf("No\n");
             }
         }
         //and
         else
         {
             scanf("%ld",&y);
             q=findnum(y);
             scanf("%s%s",b,c);
             if (strcmp(c,"siblings")==)
             {
                 if (p!=n+ && q!=n+  && p!=q && tree[p].fa==tree[q].fa) //两个结点不能相同
                     printf("Yes\n");
                 else
                     printf("No\n");
             }
             else
             {
                 scanf("%s%s%s",d,e,f);
                 if (p!=n+ && q!=n+ && tree[p].depth==tree[q].depth)
                     printf("Yes\n");
                 else
                     printf("No\n");
             }
         }
     }
     return ;
 }