TZOJ:3660: 家庭关系

时间:2021-06-24 21:03:59

描述

给定若干家庭成员之间的关系,判断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[];
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!=)
{
int num=;
map<string,int> ma;
ma.clear();
for(int i=;i<=;i++)
arr[i]=i;
for(int i=;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]==||ma[name1]==) {cout << "No" << endl;continue;}
if(find_root(ma[name])==find_root(ma[name1]))
cout << "Yes";
else cout << "No"; cout << endl;
}
return ;
}