RE 60pts求助
查看原帖
RE 60pts求助
437885
xps0606楼主2021/4/20 19:09

这是代码。

#include<bits/stdc++.h>
#define N 1100005
using namespace std;
int n,m,f[N],bian=0,sum=1;
double ans=0;
struct node{
	int ru,chu;
	double quan;
}st[N];
struct no{
	int x1,y1;
}a[N];
bool cmp(node a,node b){
	return a.quan<b.quan;
}
int find(int x){
	if(x==f[x]) return x;
	else{
		f[x]=find(f[x]);
		return f[x];
	}
}
int main()//Kruskal
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=n;i++){
		scanf("%d%d",&a[i].x1,&a[i].y1);
	}
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			int x,y;
			x=i;y=j;
			st[sum].ru=x;
			st[sum].chu=y;
			st[sum].quan=sqrt((a[x].x1-a[y].x1)*(a[x].x1-a[y].x1)+(a[x].y1-a[y].y1)*(a[x].y1-a[y].y1));
			sum++;
		}
	}
	for(int i=1;i<=m;i++){
		int x,y;
		cin>>x>>y;
		st[sum].ru=x;
		st[sum].chu=y;
		st[sum].quan=0;
		sum++;
	}
	sort(st+1,st+sum+1,cmp);
	for(int i=1;i<=sum;i++){
		int a=find(st[i].ru);
		int b=find(st[i].chu);
		if(a==b) continue;
		else{
			ans+=st[i].quan;
			f[b]=a;
			bian++;
			if(bian==n-1) break;
		}
	}
	printf("%0.2lf",ans);
	return 0;                        
}
2021/4/20 19:09
加载中...