L2-024 部落 (25分)
在一個社區里，每個人都有自己的小圈子，還可能同時屬于很多不同的朋友圈。我們認為朋友的朋友都算在一個部落里，于是要請你統計一下，在一個給定社區中，到底有多少個互不相交的部落？并且檢查任意兩個人是否屬于同一個部落。
輸入格式：

輸入在第一行給出一個正整數N（≤10
?4
?? ），是已知小圈子的個數。隨后N行，每行按下列格式給出一個小圈子里的人：
K P[1] P[2] ? P[K]
其中K是小圈子里的人數，P[i]（i=1,?,K）是小圈子里每個人的編號。這里所有人的編號從1開始連續編號，最大編號不會超過10
?4
?? 。
之后一行給出一個非負整數Q（≤10
?4
?? ），是查詢次數。隨后Q行，每行給出一對被查詢的人的編號。
輸出格式：

首先在一行中輸出這個社區的總人數、以及互不相交的部落的個數。隨后對每一次查詢，如果他們屬于同一個部落，則在一行中輸出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<iostream> #include<set> using namespace std; const int maxn = 100000; int pre[maxn]; int find(int x)//并查集模板，查找 {while (x != pre[x]) x = pre[x];return x; } void merge(int a, int b)//并查集模板，合并 {int x = find(a);int y = find(b);if (x != y) pre[x] = y; } int main() {int n, x, number,number1;for (int i = 0; i <= maxn; i++) pre[i] = i;//初始化pre數組set<int>s;//存儲人的編號，不重復cin >> n;for (int i = 0; i < n; i++){cin >> x;if (x > 0){cin >> number;s.insert(number);//將人員編號插入s，第一個人for (int j = 1; j < x; j++){cin >> number1;s.insert(number1);merge(number, number1);//合并}}}int vist[maxn]{ 0 };//標記部落是否被訪問int cnt = 0;//記錄部落個數for (auto it = s.begin(); it != s.end(); it++){int value = find(*it);pre[*it] = value;//路徑壓縮，直接與該部落首領掛鉤if (vist[value] == 0){cnt++;vist[value] = 1;}}printf("%d %d\n", s.size(), cnt);cin >> x;for (int i = 0; i < x; i++){cin >> number >> number1;if (pre[number] != pre[number1]) printf("N\n");//看是否屬于同一個部落else printf("Y\n");}}

總結

以上是生活随笔為你收集整理的C++学习之路 | PTA（天梯赛）—— L2-024 部落 (25分)（带注释）(并查集)（精简)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。