化名-并查集+二分查找

问题 A: 化名

时间限制: 1 Sec   内存限制: 128 MB
提交: 41   解决: 16
[ 提交][ 状态][ 讨论版]

题目描述

有一些间谍使用着大量不同的化名在A城活动，现在秘密警察机构需要你识别出来那些化名属于同一人，哪些化名不是。
由于经费匮乏，秘密警察机构提供给你的资料是从间谍居住区卖菜的阿呆那里搞来的。阿呆脑筋有点糊涂，他只能通过回忆一些零碎的事情来告诉你诸如：“化名p和化名q属于同一人”的事情。但阿呆和间谍们都很熟，你可以确信他提供的资料包含的信息是充分的，即不存在两个化名p和q，它们属于同一人，但根据阿呆的信息无法推断出p、q属于同一人这一事实。

输入

第1行：两个用空格分开的整数n、m，1<=n<=2000，1<=m<=20000，表示共n个化名，阿呆提供了m条信息。
第2~n+1行：每行一个长度不超过20的字符串，表示一个化名，字符串中只包含大写或小写字母。
第n+2~n+m+1行：每行二个化名p、q，中间用一个空格分开，表示阿呆告诉你p、q属于同一人。
第n+m+2行：一个整数t，1<=t<=20000，表示秘密警察机构向你提出t个问题。
第n+m+3~n+m+2+t行：每行二个化名p、q，中间用一个空格分开，表示秘密警察询问你化名p、q是否属于同一人。

输出

t行，依次对应输入文件中的t个问题。若对应问题中的两个化名属于同一人，输出“YES”，否则输出“NO”。

样例输入

5 2
a
b
c
d
e
a b
a c
2
a c
d e
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<fstream>
#include<cmath>
#include<string.h>
#include<map>
using namespace std;
int pre[2005];
map<string, int> Map;
int n, m, t;
int find(int x) {
    int r = x;
    while(pre[r] != r)
        r = pre[r];
    int i = x, j;
    while(pre[i] != r) {
        j = pre[i];
        pre[i] = r;
        i = j;
    }
    return r;   
}
 
void Mix(int x, int y) {
    int fx, fy;
    fx = find(x);
    fy = find(y);
    if(fx != fy) 
        pre[fy] = fx;       
}
 
int main() {
    char s[30], s1[30];
    while(scanf("%d%d", &n, &m) != EOF) {
        for(int i = 1; i <= n; i++) {
            scanf("%s", s);
            Map[s] = i;
            pre[i] = i;
        }
        for(int i = 1; i <= m; i++) {
            scanf("%s", s);
            scanf("%s", s1);
            Mix(Map[s], Map[s1]);
        }
        scanf("%d", &t);
        for(int i = 1; i <= t; i++) {
            scanf("%s", s);
            scanf("%s", s1);
            if(find(Map[s]) == find(Map[s1]))
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
}

样例输出

YES
NO