最小生成树80pts求助(玄2关)
查看原帖
最小生成树80pts求助(玄2关)
1287433
__ycy1124__楼主2024/7/20 16:08
#include<bits/stdc++.h>
using namespace std;
struct Node{
	int x,y,f;
}a[1001];
struct bian{
	int u,v;
	double w;
}t[1001*1001];
bool cmp(bian x1,bian x2){
	return x1.w<x2.w;
}
int find(int x){
	if(a[x].f!=x) return a[x].f=a[a[x].f].f;
	return x;
}
void hb(int x,int y){
	a[find(x)].f=find(y);
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	double ans;
	for(int i=1;i<=n;i++){
		scanf("%d%d",&a[i].x,&a[i].y);
		a[i].f=i;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			double w=1.00*sqrt(1.00*(a[i].x-a[j].x)*(a[i].x-a[j].x)+1.00*(a[i].y-a[j].y)*(a[i].y-a[j].y));
			t[(i-1)*n+j].u=i;
			t[(i-1)*n+j].v=j;
			t[(i-1)*n+j].w=w;
		}
	}
	sort(t+1,t+n*n+1,cmp);
	int js=n;
	if(k==1){
		printf("0.00");
		return 0;
	}
	for(int i=1;i<=n*n;i++){
		if(find(t[i].u)!=find(t[i].v)){
			if(js==k){
				printf("%.2lf",t[i].w);
				return 0;
			}
			hb(t[i].u,t[i].v);
			js--;
		}
	}
	return 0;
}
2024/7/20 16:08
加载中...