在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是要请你统计一下，在一个给定社区中，到底有多少个互不相交的部落？并且检查任意两个人是否属于同一个部落。

输入格式：

输入在第一行给出一个正整数N（≤104），是已知小圈子的个数。随后N行，每行按下列格式给出一个小圈子里的人：

K P[1] P[2] ⋯ P[K]

其中K是小圈子里的人数，P[i]（i=1,⋯,K）是小圈子里每个人的编号。这里所有人的编号从1开始连续编号，最大编号不会超过104。

之后一行给出一个非负整数Q（≤104），是查询次数。随后Q行，每行给出一对被查询的人的编号。

输出格式：

首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询，如果他们属于同一个部落，则在一行中输出Y，否则输出N。

输入样例：

1
2
3
4
5
6
7
8
4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7

输出样例：

1
2
3
10 2
Y
N

思路：

  • 并查集应用，同一个部落的进行合并，固定模板即可，祖先是自己的，说明是一个部落
  • 采用了set来存储社区中人的编号，既为每次输入去了重，又解决了社区总人数问题，因为题目中说编号从1开始连续编号，则最大编号者即为总人数。

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#include <iostream>
#include <set>
using namespace std;

int f[10001];//并查集数组

void init() {
    for (int i = 1; i <= 10000; i++) {
        f[i] = i;
    }
}

int find(int x) {
    if (x == f[x])
        return x;
    else//路径压缩，寻找祖先的同时更改上层父结点值
        return f[x] = find(f[x]);
}

void merge(int x, int y) {
    int a = find(x);
    int b = find(y);
    if (a != b)
        f[b] = a;
}

int main() {
    int n, k, first, id;
    cin >> n;
    init();//并查集数组初始化
    set<int> st;
    for (int i = 1; i <= n; i++) {
        cin >> k >> first;
        st.insert(first);
        for (int j = 1; j < k; j++) {
            cin >> id;
            st.insert(id);
            merge(id, first);
        }
    }
    int cnt = 0;
    for (int i = 1; i <= st.size(); i++) {
        if (f[i] == i)
            cnt++;
    }
    cout << st.size() << " " << cnt << endl;
    int q, t1, t2;
    cin >> q;
    while (q--) {
        cin >> t1 >> t2;
        if (find(t1) == find(t2))
            cout << "Y\n";
        else
            cout << "N\n";
    }
    return 0;
}