AcWing 1145. 北极通讯网络

题目

北极的某区域共有 n 座村庄，每座村庄的坐标用一对整数 (x,y) 表示。

为了加强联系，决定在村庄之间建立通讯网络，使每两座村庄之间都可以直接或间接通讯。

通讯工具可以是无线电收发机，也可以是卫星设备。

无线电收发机有多种不同型号，不同型号的无线电收发机有一个不同的参数 d，两座村庄之间的距离如果不超过 d，就可以用该型号的无线电收发机直接通讯，d 值越大的型号价格越贵。现在要先选择某一种型号的无线电收发机，然后t统一给所有村庄配备，数量不限，但型号都是 相同的。

配备卫星设备的两座村庄无论相距多远都可以直接通讯，但卫星设备是 有限的，只能给一部分村庄配备。

现在有 k 台卫星设备，请你编一个程序，计算出应该如何分配这 k 台卫星设备，才能使所配备的无线电收发机的 d 值最小。

例如，对于下面三座村庄：

其中，|AB|=10,|BC|=20,|AC|=10√5≈22.36。

如果没有任何卫星设备或只有 1 台卫星设备 (k=0 或 k=1)，则满足条件的最小的 d=20，因为 A 和 B，B 和 C 可以用无线电直接通讯；而 A 和 C 可以用 B 中转实现间接通讯 (即消息从 A 传到 B，再从 B 传到 C)；

如果有 2 台卫星设备 (k=2)，则可以把这两台设备分别分配给 B 和 C ，这样最小的 d 可取 10，因为 A 和 B 之间可以用无线电直接通讯；B 和 C 之间可以用卫星直接通讯；A 和 C 可以用 B 中转实现间接通讯。

如果有 3 台卫星设备，则 A,B,C 两两之间都可以直接用卫星通讯，最小的 d 可取 0。

输入格式

第一行为由空格隔开的两个整数 n,k;

接下来 n 行，每行两个整数，第 i 行的 xi,yi 表示第 i 座村庄的坐标 (xi,yi)。

输出格式

一个实数，表示最小的 d 值，结果保留 2 位小数。

数据范围

1≤n≤500,
0≤x,y≤10^4,
0≤k≤100

思路

跑一遍最小生成树，删去最大的k跳边即可。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;

#define x first
#define y second
const int N=510,M=N*N/2;
int n,k,m;
struct Edge
{
	int a,b;
	double w;
	bool operator<(const Edge &t) const
	{
		return w<t.w;
	}
}e[M];

pii q[M];
int p[N];

double get_dist(pii a, pii b)
{
	int dx = a.x - b.x;
	int dy= a.y -b .y;
	return sqrt(dx*dx +dy*dy);
}

int find(int x)
{
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}

int main()
{
	//freopen("test.in","r",stdin);//设置 cin scanf 这些输入流都从 test.in中读取
    //freopen("test.out","w",stdout);//设置 cout printf 这些输出流都输出到 test.out里面去
	//ios::sync_with_stdio(false);
	//cin.tie(0),cout.tie(0);
	cin>>n>>k;
	for(int i=0;i<n;i++) cin>>q[i].x>>q[i].y;
	for(int i=0;i<n;i++)
		for(int j=0;j<i;j++)
			e[m++]={i,j,get_dist(q[i],q[j])};
	sort(e,e+m);
	for(int i=0;i<n;i++) p[i]=i;
	int cnt=n;
	double res=0;
	for(int i=0;i<m;i++)
	{
		if(cnt<=k) break;
		int a=find(e[i].a),b=find(e[i].b);
		double w=e[i].w;
		if(a!=b)
		{
			p[a]=b;
			cnt--;
			res=w;
		}
	}
	printf("%.2lf\n",res);
	return 0;
}