L3-1 社交叢集(30 分)
阿新 • • 發佈:2019-02-01
當你在社交網路平臺註冊時,一般總是被要求填寫你的個人興趣愛好,以便找到具有相同興趣愛好的潛在的朋友。一個“社交叢集”是指部分興趣愛好相同的人的集合。你需要找出所有的社交叢集。
輸入格式:
輸入在第一行給出一個正整數 N(≤1000),為社交網路平臺註冊的所有使用者的人數。於是這些人從 1 到 N 編號。隨後 N 行,每行按以下格式給出一個人的興趣愛好列表:
Ki: hi[1] hi[2] ... hi[Ki]
其中Ki(>0)是興趣愛好的個數,hi[j]是第j個興趣愛好的編號,為區間 [1, 1000] 內的整數。
輸出格式:
首先在一行中輸出不同的社交叢集的個數。隨後第二行按非增序輸出每個叢集中的人數。數字間以一個空格分隔,行末不得有多餘空格。
輸入樣例:
8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4
輸出樣例:
3
4 3 1
很明顯是並查集,比賽的時候我在想怎麼判斷兩個人有相同的興趣,,結果還是暴力。。先暴力搜尋有相同興趣的兩個人,剩下的就是並查集了
#include <bits/stdc++.h> using namespace std; const int N = 1e3 + 3; int f[N], ch[N]; int Find(int x) { if(f[x] != x) f[x] = Find(f[x]); return f[x]; } void Union(int x, int y) { int fx = Find(x); int fy = Find(y); if(fx != fy) { f[fx] = fy; ch[fy] += ch[fx]; } } int main() { int n, k, h; scanf("%d", &n); vector<set<int> > vec(N); for(int i = 1; i <= n; i++) { f[i] = i; ch[i] = 1; } for(int i = 1; i <= n; i++) { scanf("%d:", &k); while(k--) { scanf("%d", &h); vec[i].insert(h); for(int j = 1; j < i; j++) { if(vec[j].count(h) != 0) { Union(i, j); break; } } } } int cnt = 0, ans[N]; for(int i = 1; i <= n; i++) { if(Find(i) == i) { ans[cnt] = ch[i]; cnt++; } } sort(ans, ans + cnt); printf("%d\n", cnt); for(int i = cnt-1; i >= 0; i--) { printf("%d", ans[i]); if(i > 0) printf(" "); } return 0; }