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

题目1035：找出直系亲属

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：1309

解决：521

题目描述：

如果A，B是C的父母亲，则A，B是C的parent，C是A，B的child，如果A，B是C的（外）祖父，祖母，则A，B是C的grandparent，C是A，B的grandchild，如果A，B是C的（外）曾祖父，曾祖母，则A，B是C的great-grandparent，C是A，B的great-grandchild，之后再多一辈，则在关系上加一个great-。

输入：

输入包含多组测试用例,每组用例首先包含2个整数n（0<=n<=26）和m（0<m<50）, 分别表示有n个亲属关系和m个问题, 然后接下来是n行的形式如ABC的字符串，表示A的父母亲分别是B和C，如果A的父母亲信息不全，则用-代替，例如A-C,再然后是m行形式如FA的字符串,表示询问F和A的关系。
当n和m为0时结束输入。

输出：

如果询问的2个人是直系亲属，请按题目描述输出2者的关系，如果没有直系关系，请输出-。
具体含义和输出格式参见样例.

样例输入：

3 2
ABC
CDE
EFG
FA
BE
0 0

样例输出：

great-grandparent
-

来源：

2009年浙江大学计算机及软件工程研究生机试真题

MYCode

#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;
        }
    }
}

//status：accepted

根据题目给的数据构建一棵二叉树

比如 题目给出的数据可以构成下面的二叉树

  A

↙↘

BC

↙↘

D E

    ↙↘

   F     G

判断F和A是否具有直系关系

从F一直向上搜索

F到E到C到A

找到了A说明F和A具有直属关系。

否则不具有直属关系。