[Jobdu] 题目1035：找出直系亲属

 

题目描述：

    如果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
-

 

用并查集实现，代码如下：

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <map>
using namespace std;

map<char,char> fa;
char x[3];
int m,n;

inline int max(int a,int b){ return a-b; }

int find(char a,char b){
    int i = 0;
    char c = a;
    bool flag = false;
    while(c!=fa[c]){
        c = fa[c];
        i++;
        if(c==b){
            flag = true;
            break;
        }
    }
    if(flag){
        return i;
    }
    return 0;
}

int main(int argc,char* argv[]){
    //freopen("input.txt","r",stdin);
    while(cin>>n>>m&&n&&m){
        fa.clear();
        for(int i=0;i<n;i++){
            scanf("%s",&x);
            for(int j=1;j<3;j++){
                if(x[i]!='-'){
                    fa[x[j]] = x[0];
                }
            }
        }
        for(int i=0;i<m;i++){
            scanf("%s",&x);
            int c = max(find(x[0],x[1]),find(x[1],x[0]));
            if(c==0){
                printf("-\n");
            }
            else{
                for(int k=0;k<abs(c)-1;k++){
                    if(k==abs(c)-2){
                        printf("grand");
                        break;
                    }
                    printf("great-");
                }
                if(c>0){
                    printf("parent\n");
                }
                else{
                    printf("child\n");
                }
            }
        }
    }
    return 0;
}