Bear and Friendship Condition CodeForces - 771A（并查集判断完全图，好题）

题意：有一个无向图，让判断是不是所有的点都满足：如果A和B连接并且C和A连接，那么C和A连接。参考博客

题解：显然这是一个完全图，而n个点的无向完全图有n*(n-1)/2条边，利用并查集统计连通块边数和点数，最后进行判断就行了。

AC代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
ll n,m,f[maxn],bian[maxn],dian[maxn];
void init() {
	for(int i=1; i<=n; i++) {
		f[i]=i;
		dian[i]=1;
		bian[i]=0;
	}
}
int find_x(int x) {
	if(x!=f[x])return f[x]=find_x(f[x]);
	return x;
}
void unite(int a,int b) {
	ll aa=-1,bb=-1;
	 aa=find_x(a);
	 bb=find_x(b);
	if(aa!=bb) {
		f[aa]=bb;
		bian[bb]+=bian[aa]+1;
		dian[bb]+=dian[aa];
	} else bian[aa]++;
}
bool check(){
	for(int i=1;i<=n;i++){
		if(f[i]!=i)continue;//只找每个连通块的根节点判断
		if(bian[i]!=(dian[i]-1)*dian[i]/2)return false; 
	}
	return true;
}
int main() {
	cin>>n>>m;
	init();
	for(int i=1; i<=m; i++) {
		int a,b;
		cin>>a>>b;
		unite(a,b);
	}
	//cout<<123<<endl;
    if(check())cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
}