站外梯
  • 板块灌水区
  • 楼主lucy2012
  • 当前回复0
  • 已保存回复0
  • 发布时间2024/7/19 16:44
  • 上次更新2024/7/19 17:35:05
查看原帖
站外梯
1252442
lucy2012楼主2024/7/19 16:44

改用并查集做了做,样例未过,题目。

代码(我在原帖讨论了里也发了):

#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;
}
2024/7/19 16:44
加载中...