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;
}