洛谷P2256 一中校运会之百米跑

题目背景

在一大堆秀恩爱的**之中，来不及秀恩爱的苏大学神踏着坚定（？）的步伐走向了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."。

输入输出样例

输入 #1

10 6
Jack
Mike
ASDA
Michel
brabrabra
HeHe
HeHE
papapa
HeY
Obama
Jack Obama
HeHe HeHE
brabrabra HeHe
Obama ASDA
papapa Obama
Obama HeHE
3
Mike Obama
HeHE Jack
papapa brabrabra

输出 #1

No.
Yes.
Yes.

Solution

和并查集的模板题差不多

这次我还用了按秩合并！

以及

STL大法好！

Code

 1 #include<iostream>
 2 #include<algorithm>
 3 #include<map>
 4 using namespace std;
 5 map<string,int>name;
 6 int n,m,k,f[5001],size[5001],ans,yet;
 7 int getf(int a)
 8 {
 9     if(f[a]==a) return a;
10     return f[a]=getf(f[a]);//路径压缩 
11 }
12 void merge(int a,int b)
13 {
14     int i=getf(a);
15     int j=getf(b);
16     if(size[i]>size[j]) swap(i,j);
17     f[j]=i;
18     size[i]+=size[j];
19 }
20 int main()
21 {
22     cin>>n>>m;
23     string str,str1,str2;
24     for(int i=1;i<=n;i++)
25     {
26         size[i]=1;
27         f[i]=i;
28     }
29     for(int i=1;i<=n;i++)
30     {
31         cin>>str;
32         name[str]=i;
33     }
34     for(int i=1;i<=m;i++){
35         cin>>str1>>str2;
36         merge(name[str1],name[str2]);
37     }
38     cin>>k;
39     for(int i=1;i<=k;i++)
40     {
41         cin>>str1>>str2;
42         if(getf(name[str1])==getf(name[str2])) cout<<"Yes.\n";
43         else cout<<"No.\n";
44     }
45     return 0;
46 }