题目背景
在一大堆秀恩爱的**之中,来不及秀恩爱的苏大学神踏着坚定(?)的步伐走向了100米跑的起点。这时苏大学神发现,百米赛跑的参赛同学实在是太多了,连体育老师也忙不过来。这时体育老师发现了身为体育委员的苏大学神,便来找他帮忙。可是苏大学神需要热身,不然跑到一半就会抽(筋)、于是他就找到了你。。。如果你帮助体育老师解决了问题,老师就会给你5个积分。
题目描述
假设一共有N(2<=N<=20000)个参赛选手。(尼玛全校学生都没这么多吧)
老师会告诉你这N个选手的名字。
接着会告诉你M(1<=M<=1000000)句话,即告诉你学生A与学生B在同一个组里。
如果学生A与学生B在同一组里,学生B与学生C也在同一组里,就说明学生A与学生C在同一组。
然后老师会问你K(1<=K<=1000000)句话,即学生X和学生Y是否在同一组里。
若是则输出"Yes.",否则输出"No."
输入格式
第一行输入N和M。
接下来N行输入每一个同学的名字。
再往下M行每行输入两个名字,且保证这两个名字都在上面的N行中出现过,表示这两个参赛选手在同一个组里。
再来输入K。
接下来输入K个体育老师的询问。
输出格式
对于每一个体育老师的询问,输出"Yes."或"No."。
这道题的题目描述很奇怪QWQ。
简单翻译一下:先给你N个同学,告诉你他们的名字分别是什么。然后M行每行两个名字,代表这两个人在同一个组里。最后K行是K组询问,每组询问有两个名字,代表询问这两个人是否在同一个组里。
那么这道题就很明显是一个并查集的简单应用了。只是与一般的并查集不同的是,这里的元素下标不是数字,而是字符串,所以我们首先需要使用一个映射map将每个名字映射到一个数字上。这里我将名字映射到了读入元素的下标上。
映射构造代码如下:
1 #include<iostream>
2 #include<map>
3 #include<string>
4 using namespace std;
5 int n,m;
6 string s;
7 string s1,s2;
8 map <string,int> h;
9 int main(){
10 cin>>n>>m;
11 for(int i=1;i<=n;i++){
12 cin>>s;
13 h[s]=i;
14 }
15 return 0;
16 }
这样我们在进行合并和查询时就只需要将h[s]作为元素下标即可。
接下来将并查集的模板稍加修改即可。
完整代码如下:
1 #include<iostream>
2 #include<map>
3 #include<string>
4 using namespace std;
5 int n,m,k;
6 int father[20005];
7 string s;
8 string s1,s2;
9 map <string,int> h;
10 int find(int x){
11 if(father[x]!=x){
12 return father[x]=find(father[x]);
13 }else{
14 return x;
15 }
16 }
17 void unionn(int r1,int r2){
18 int x=find(r1);
19 int y=find(r2);
20 father[x]=y;
21 }
22 int main(){
23 cin>>n>>m;
24 for(int i=1;i<=n;i++){
25 cin>>s;
26 h[s]=i;
27 father[h[s]]=i;
28 }
29 for(int i=1;i<=m;i++){
30 cin>>s1>>s2;
31 unionn(h[s1],h[s2]);
32 }
33 cin>>k;
34 for(int i=1;i<=k;i++){
35 cin>>s1>>s2;
36 if(find(h[s1])==find(h[s2])){
37 cout<<"Yes."<<endl;
38 }else{
39 cout<<"No."<<endl;
40 }
41 }
42 return 0;
43 }