找出直系亲属【并查集、vector路径存储】

2023-05-25 15:05:12 浏览数 (2)

这个题 找父亲发现不唯一,但是找儿子确是唯一的,那么可以倒过来建树

描述

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

输出描述:

对于每组输入数据,如果所有顶点都是连通的,输出"YES",否则输出"NO"。

示例

输入: 4 3 1 2 2 3 3 2 3 2 1 2 2 3 0 0 输出: NO YES

参考代码:

路径查找的代码封装成函数更好,我这里重复了两遍

代码语言:javascript复制
#include<iostream>
#include<map>
#include<vector>
using namespace std;
int son[30];
int height[30];//其实没用到
int n,m;
int s;
vector<int>v;
void Initial()
{
    for(int i=0; i<=27; i  )
    {
        son[i]=i;
        height[i]=0;

    }
}

int Find(int x)
{
    s  ;
//    cout<<x<<" has a son is "<<son[x]<<endl;
    v.push_back(x);
    if(x!=son[x])
    {
        return Find(son[x]);
    }
    return son[x];
}
void Union(int x,int y)//这个题每个节点关系很明确,后面的是父母,倒过来建树父母的儿子就是前者,不需要进行常规的height判断
{
    son[y]=x;
    height[x]  ;
    return ;
}
bool is_parents(char p)
{
    if(p>='A'&&p<='Z')return true;
    return false;
}

void relation_find(int p1,int p2)
{
    s=0;
    int a =Find(p1);
    int d1 =s;
//    cout<<a<<" "<<s<<endl;
    s=0;
    int b=Find(p2);
    int d2 =s;
//    cout<<b<<" "<<s<<endl;
    if(a !=b)//根源不同
    {
        cout<<"-"<<endl;
    }
    else
    {

        if(d1<d2) //d2更大,说明 p2走过的路径深,是p1的长辈
        {
            v.clear();
            Find(p2);//打印p2路径
            int f=0;
            for(int i=0; i<v.size(); i  )
            {
                if(v[i]==p1)f=1;//p1在p2走过的路径中,说明是直系
            }

            if(f)
            {
                int d = d2-d1;
                if(d == 1)cout<<"child"<<endl;
                else if(d ==2)cout<<"grandchild"<<endl;
                else
                {
                    string great="";
                    d =d-2;
                    while(d--)great ="great-";
                    great ="grandchild";
                    cout<<great<<endl;
                }
            }
            else cout<<"-"<<endl;
        }
        else
        {

            v.clear();
            Find(p1);
            int f=0;
            for(int i=0; i<v.size(); i  )
            {
                if(v[i]==p2)f=1;
            }

            if(f)
            {
                int d = d1-d2;
                if(d == 1)cout<<"parent"<<endl;
                else if(d ==2)cout<<"grandparent"<<endl;
                else
                {
                    string great="";
                    d =d-2;
                    while(d--)great ="great-";
                    great ="grandparent";
                    cout<<great<<endl;
                }
            }
            else cout<<"-"<<endl;

        }



    }
}
int main()
{
    while(cin>>n>>m)
    {
//        cout<<n<<" "<<m<<endl;
        Initial();
//        cout<<n<<endl;
        while(n--)
        {
            string str;
            cin>>str;
            int son = str[0]-'A';
            int f1 = str[1]-'A';
            int f2= str[2]-'A';
            if(is_parents(str[1]))Union(son,f1);
            if(is_parents(str[2]))Union(son,f2);
        }
        while(m--)
        {
            string str;
            cin>>str;
            int p1 = str[0]-'A';
            int p2 = str[1]-'A';
            relation_find(p1,p2);
        }
    }
}

0 人点赞