AtCoder Beginner Contest 157 D Friend Suggestions 容斥原理+并查集

AtCoder Beginner Contest 157   比赛人数7464   与codeforces比赛冲突，遗憾没有参加实时赛，之后模拟赛，打得没劲

AtCoder Beginner Contest 157 D Friend Suggestions  容斥原理+并查集

总目录详见https://blog.csdn.net/mrcrack/article/details/104454762

在线测评地址https://atcoder.jp/contests/abc157/tasks/abc157_d

思路:答案=统计某人在同一个并查集中其他人数-该人对应的friendship人数-在该并查集中该人对应的blockship人数

#include <stdio.h>
#define maxn 100010
int n,m,k,f[maxn],fn[maxn],cnt[maxn],tot[maxn],ans[maxn];;
int getf(int u){//找爸爸
	return f[u]==u?u:f[u]=getf(f[u]);
}
void merge(int u,int v){//合并并查集
	int f1=getf(u),f2=getf(v);
	if(f1!=f2)f[f2]=f1;//左靠
}
int main(){
	int i,j,a,b;
	scanf("%d%d%d",&n,&m,&k);
	for(i=1;i<=n;i++)f[i]=i;
	for(i=1;i<=m;i++){
		scanf("%d%d",&a,&b);
		cnt[a]++,cnt[b]++,merge(a,b);//cnt[i]用来统计i的friendship数量
	}
	for(i=1;i<=n;i++)f[i]=getf(i);//更新父亲
	for(i=1;i<=n;i++)fn[f[i]]++;//统计同一并查集中人员数量
	for(i=1;i<=k;i++){
		scanf("%d%d",&a,&b);
		if(f[a]==f[b])tot[a]++,tot[b]++;//tot[i]统计在i所在并查集中，与i有blockship关系的人员数量
	}
	for(i=1;i<=n;i++)ans[i]=fn[f[i]]-cnt[i]-tot[i]-1;//1是指i自己,所有需扣除1
	for(i=1;i<n;i++)printf("%d ",ans[i]);
	printf("%d\n",ans[n]);
	return 0;
}