CodeForces-766D Mahmoud and a Dictionary 并查集 维护同类不同类元素集合
题目链接:https://cn.vjudge.net/problem/CodeForces-766D
题意
写词典,有些词是同义词,有些是反义词,还有没关系的词
首先输入两个词,需要判断是同义还是是反义,若没关系就按操作归为同义或反义
思路
经典并查集的动物园问题
维护两个并查集,find(a)表示a的同类代表元,find(a+maxn)表示异类代表元
find(a)find(b) && find(a+maxn)find(b+maxn) 就是说ab同类
find(a+maxn)find(b) && find(a)find(b+maxn) 就是说ab异类
其他就是没关系,需要合并
提交过程
WA | 输出问题,NO写成ON,老毛病 |
AC |
代码
#include <map>
#include <cstdio>
#include <string>
#include <cstring>
using namespace std;
const int maxn=1e5+20;
int pre[maxn*2];
map<string, int> toid;
int find(int x){
return (x==pre[x])?x:(pre[x]=find(pre[x]));
}
int join(int a, int b){
a=find(a); b=find(b);
if (a!=b) pre[a]=b;
}
int main(void){
int n, m, q;
char str[100], bstr[100];
while (scanf("%d%d%d", &n, &m, &q)==3 && n){
toid.clear();
for (int i=1; i<=n; i++){
scanf("%s", str);
toid[string(str)]=i;
}
int arg, a, b;
for (int i=1; i<=maxn*2; i++) pre[i]=i;
for (int i=0; i<m; i++){
scanf("%d%s%s", &arg, str, bstr);
a=toid[string(str)];
b=toid[string(bstr)];
if (find(a)==find(b) && find(a+maxn)==find(b+maxn)){
if (arg==1) printf("YES\n");
else printf("NO\n");
}else if (find(a)==find(b+maxn) && find(a+maxn)==find(b)){
if (arg==2) printf("YES\n");
else printf("NO\n");
}else{
printf("YES\n");
if (arg==1) {join(a, b); join(a+maxn, b+maxn);}
else {join(a, b+maxn); join(a+maxn, b);}
}
}
while (q--){
scanf("%s%s", str, bstr);
a=toid[string(str)];
b=toid[string(bstr)];
if (find(a)==find(b) && find(a+maxn)==find(b+maxn)){
printf("1\n");
}else if (find(a)==find(b+maxn) && find(a+maxn)==find(b)){
printf("2\n");
}else printf("3\n");
}
}
return 0;
}
Time | Memory | Length | Lang | Submitted |
---|---|---|---|---|
732ms | 8380kB | 1328 | GNU G++ 5.1.0 | 2018-07-28 14:45:06 |