找直系亲属——并查集

题目描述

    如果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的关系。
    

输出描述:

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

输入

复制
3 2
ABC
CDE
EFG
FA
BE

输出

复制
great-grandparent

-

//用并查集,并查集可以找到某结点的根节点
//该题孩子是独生子女,可以构建一棵树,即双亲的根节点是孩子
//如果某个结点不断的追溯其根节点可以到达另一个结点那么是直系亲属
//否则就不是直系,在追溯的过程中记录下来追溯的次数决定输出几个great
#include<iostream>
#include<string.h>
using namespace std;
int pre[100];
int getindex(int x,int y)
{
    //如果x是孩子,y是长辈
    int count=1;
    //int t=y;
    while(y!=-1)
    {
        if(pre[y]==x)
            return count;
        else{
            y=pre[y];
            count++;
        }
    }
    return -1;
}
int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        string s;
        memset(pre,-1,sizeof(pre));
        while(n--)
        {
            cin>>s;
            for(int i=1;i<s.size();i++)
            {
                if(s[i]!='-')
                      pre[s[i]-'A']=s[0]-'A';//把长辈的根节点设置为孩子的编号。
            }
        }
        char a,b;
        while(m--)
        {
            cin>>a>>b;
            int index1=getindex(a-'A',b-'A');//得到追溯的层数
            int index2=getindex(b-'A',a-'A');//得到追溯的层数
            if(index1==-1&&index2==-1)cout<<"-"<<endl;
            else if(index1!=-1)//说明a是孩子
           {
                if(index1==1)cout<<"child"<<endl;
                else if(index1==2)cout<<"grandchild"<<endl;
                else{
                    for(int i=2;i<index1;i++)
                   {
                        cout<<"great-";
                   }
                        cout<<"grandchild"<<endl;
                }
            }
            else if(index2!=-1)//说明a是孩子
            {
                if(index2==1)cout<<"parent"<<endl;
                else if(index1==2)cout<<"grandparent"<<endl;
                else{
                     for(int i=2;i<index2;i++)
                    {
                       cout<<"great-";
                    }
                      cout<<"grandparent"<<endl;
                }
            }

        }


    }

}

猜你喜欢

转载自blog.csdn.net/qq_38030194/article/details/80872845