改用并查集做了做,样例未过,题目。
代码(我在原帖讨论了里也发了):
#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;
}