九度OJ 1035:找出直系亲属(二叉树)

#include<iostream>
#include<cstring>
#include<cstdio>
using  namespace std;
#define MAX  110
int pre[ 27];
int find( int id1,  int id2)
{
     int  ct =  0;
     while(id1 != id2 && id1 != - 1)
    {
        id1 = pre[id1];
        ct++;
    }
     if(id1 == id2)
         return ct;
     return  0;
}
int main()
{
     int n, m;
     while(scanf( "%d%d", &n, &m) != EOF)
    {
         if(n ==  0 && m ==  0)
             break;
        memset(pre, - 1,  sizeof(pre));
         char str[ 3];
         int i;
         for(i =  1; i <= n; i++)
        {
            scanf( "%s", &str);
             int a, b = - 1, c = - 1;
            a = str[ 0] -  'A';
             if(str[ 1] >=  'A' && str[ 1] <=  'Z')
                b = str[ 1] -  'A';
             if(str[ 2] >=  'A' && str[ 2] <=  'Z')
                c = str[ 2] -  'A';
             if(b != - 1)
                pre[b] = a;
             if(c != - 1)
                pre[c] = a;
        }
         /*for(i=0;i<26;i++)
        cout<<pre[i]<<" ";
        cout<<endl;*/
         for(i =  1; i <= m; i++)
        {
             char ch1, ch2;
            cin >> ch1 >> ch2;
             int index1 = ch1 -  'A';
             int index2 = ch2 -  'A';
             bool flag =  false;
             int res1 = find(index2, index1);
             //cout<<"res1="<<res1<<endl;
             if(res1)
            {
                flag =  true;
                 switch(res1)
                {
                 case  1:
                    cout <<  "child" << endl;
                     break;
                 case  2:
                    cout <<  "grandchild" << endl;
                     break;
                 default:
                     for( int k = res1; k >=  3; k--)
                        cout <<  "great-";
                    cout <<  "grandchild" << endl;
                     break;
                }
            }
             else
            {
                 int res2 = find(index1, index2);
                 //cout<<"res2="<<res2<<endl;
                 if(res2)
                {
                    flag =  true;
                     switch(res2)
                    {
                     case  1:
                        cout <<  "parent" << endl;
                         break;
                     case  2:
                        cout <<  "grandparent" << endl;
                         break;
                     default:
                         for( int k = res2; k >=  3; k--)
                            cout <<  "great-";
                        cout <<  "grandparent" << endl;
                         break;
                    }
                }
            }
             if(!flag)
                cout <<  "-" << endl;
        }
    }
}