在一个社区里,每个人都有自己的小圈子,还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里,于是要请你统计一下,在一个给定社区中,到底有多少个互不相交的部落?并且检查任意两个人是否属于同一个部落。
输入格式:
输入在第一行给出一个正整数
K P[1] P[2] ⋯ P[K]
其中K是小圈子里的人数,P[i](i=1,⋯,K)是小圈子里每个人的编号。这里所有人的编号从1开始连续编号,最大编号不会超过
之后一行给出一个非负整数
输出格式:
首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询,如果他们属于同一个部落,则在一行中输出Y,否则输出N。
输入样例:
4 3 10 1 2 2 3 4 4 1 5 7 8 3 9 6 4 2 10 5 3 7
输出样例:
10 2 Y N
分析:
并查集简单题
代码:
#include <bits/stdc++.h> #define LL long long using namespace std; const int maxn = 1e4 + 10; const int inf = 0x3f3f3f3f; const double PI = acos(-1.0); typedef pair<int, int> PII; int vis[maxn], fa[maxn]; int f(int x) { return fa[x] == x ? x : fa[x] = f(fa[x]); } int main(int argc, char const *argv[]) { for (int i = 0; i < maxn; i++) fa[i] = i; int n; cin >> n; while (n--) { int k; cin >> k; int x; cin >> x; vis[x]++; k--; int root = f(x); while (k--) { cin >> x; vis[x]++; fa[f(x)] = root; } } int sum = 0, cnt = 0; for (int i = 1; i < maxn; i++) { if (vis[i] != 0) { sum++; if (f(i) == i) cnt++; } } cout << sum << ' ' << cnt << endl; int q; cin >> q; while (q--) { int x, y; cin >> x >> y; if (f(x) == f(y)) puts("Y"); else puts("N"); } return 0; }