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

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

描述

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

参考代码：

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

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