News Distribution（Codeforces 1167C） （并查集简单应用）

并查集查询时间复杂度是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;
}