TZOJ：3660: 家庭关系

描述

给定若干家庭成员之间的关系，判断2个人是否属于同一家庭，即2个人之间均可以通过这些关系直接或者间接联系。

输入

输入数据有多组，每组数据的第一行为一个正整数n(1<=n<=100)，表示有100个关系描述，接下来有n行，每行的描述方式为：
p1 p2 c
其中p1、p2和c均为一串文本，表示每个人的姓名，p1和p2为c的父亲和母亲。
最后一行包含2个字符串a和b，为待判断的两个人的姓名。
每个人的姓名由大小写字母组成，长度不超过80。

若n为0，表示输入结束。

输出

如果a和b在同一个家庭中，则输出Yes
否则输出No

样例输入

2 Barbara Bill Ted Nancy Ted John John Barbara 3 Lois Frank Jack Florence Bill Fred Annie Fred James James Jack 0

样例输出

Yes No

题目来源

TZOJ

解题思路：映射一下 然后并查集就好了

菜鸡的成长史 ^_^

 

#include <bits/stdc++.h>
using namespace std;
int arr[505];
int find_root(int x) {return arr[x]==x?x:arr[x]=find_root(arr[x]);}
void unionset(int x,int y)
{
    int xx=find_root(x),yy=find_root(y);
    if(xx!=yy) arr[xx]=yy;
}
int main()
{
    ios::sync_with_stdio(false);
    int n,m;
    string s,s1,s2,name,name1;
    while(cin>>n&&n!=0)
    {
        int num=0;
        map<string,int> ma;
        ma.clear();
        for(int i=1;i<=505;i++)
            arr[i]=i;
        for(int i=1;i<=n;i++){
            cin>>s>>s1>>s2;  //给字符编一个号
            if(!ma[s]) ma[s]=++num;
            if(!ma[s1]) ma[s1]=++num;
            if(!ma[s2]) ma[s2]=++num;
            unionset(ma[s],ma[s1]),unionset(ma[s],ma[s2]);
        }
        cin>>name>>name1;
        if(ma[name]==0||ma[name1]==0) {cout << "No" << endl;continue;}
        if(find_root(ma[name])==find_root(ma[name1]))
            cout << "Yes";
        else cout << "No"; cout << endl;
    }
    return 0;
}