这个并查集超时加wa,应该不是正解,先求调,如果调不了求思路:
#include<bits/stdc++.h>
using namespace std;
int n,m,a[100010],vis[100010],tog[100010],cnt;
long long sum=0;
int find(int x){
if(a[x]==x)
return x;
return a[x]=find(a[x]);
}
void toge(int x,int y){
int xb=find(x),yb=find(y);
a[xb]=yb;
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
a[i]=i;
}
for(int i=1;i<=m;i++){
int u,v;
cin>>u>>v;
toge(u,v);
}
for(int i=1;i<=n;i++){
if(!vis[a[i]]){
cnt++;
tog[cnt]=1;
vis[a[i]]=cnt;
}else{
tog[vis[a[i]]]++;
}
}
for(int i=1;i<=cnt;i++){
for(int j=i+1;j<=cnt;j++){
sum+=tog[i]*tog[j];
}
}
cout<<sum;
return 0;
}