洛谷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."。
输入输出样例
输入 #110 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 }