题目背景
在一大堆秀恩爱的**之中,来不及秀恩爱的苏大学神踏着坚定(?)的步伐走向了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 }