并查集查询时间复杂度是O(1),合并时间复杂度才是O(n)
题意:n(人数),m(组数),m行,先输入k,表示这组有k个人,下面是k个人的编号,同组可以传递信息,问当第i个人是信息源时,有几个人知道信息。
AC代码:
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <algorithm>
#define IOS ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
const int maxn=1e6+5;
int n,m,f[maxn],num[maxn];
int find_x(int x) {
if(f[x]!=x)return f[x]=find_x(f[x]);//这个地方必须有f[x]=否者会超时
else return f[x];//这个地方必须有return 否者会RE
}
void unite(int x,int y) {
int aa=find_x(x);
int bb=find_x(y);
if(aa!=bb)f[aa]=bb;
}
int main() {
IOS;
cin>>n>>m;
for(int i=0; i<=n; i++)f[i]=i;
while(m--){
int k,a,b;
cin>>k;
if(k!=0)cin>>a;
for(int i=1; i<k; i++) {
cin>>b;
unite(a,b);
}
}
for(int i=1; i<=n; i++)num[find_x(i)]++;
cout<<num[find_x(1)];
for(int i=2; i<=n; i++)cout<<" "<<num[find_x(i)];
cout<<endl;
}