【Kruskal】北极通讯网络

题目描述

北极的某区域共有n座村庄（1≤n≤500），每座村庄的坐标用一对整数（x，y）表示，其中0≤x，y≤10000。为了加强联系，决定在村庄之间建立通讯网络。通讯工具可以是无线电收发机，也可以是卫星设备。所有的村庄都可以拥有一部无线电收发机， 且所有的无线电收发机型号相同。但卫星设备数量有限，只能给一部分村庄配备卫星设备。 

不同型号的无线电收发机有一个不同的参数d，两座村庄之间的距离如果不超过d就可以用该型号的无线电收发机直接通讯，d值越大的型号价格越贵。拥有卫星设备的两座村庄无论相距多远都可以直接通讯。

现在有k台（1≤k≤100）卫星设备，请你编写一个程序，计算出应该如何分配这k台卫星设备，才能使所有的无线电收发机的d值最小，并保证每两座村庄之间都可以直接或间接地通讯。

输入

第一行包括两个整数n、k，表示村庄的数量和卫星设备的数量。

之后的n行，输入xi，yi，表示第i个村庄的坐标。

输出

输出一个数，代表d的最小值。输出保留两位小数。

样例输入

3 2
10 10
10 0
30 0

样例输出

10.00

题解

这题是就是用最小生成树的Kruskal方法来做，为使所有村庄可以通信，将村庄都看作一个点，需要连接的边的条数等于村庄的数量减去卫星设备的数量，即n-k（有卫星设备的点看作是已经选中的点，没有的点是待加入的点），由于Kruskal加入边是按照由小到大的顺序进行的，所以最后一次加入的边的长度就是最小的d值。

#include<iostream>
#include<algorithm>
#include<cmath>
#include<iomanip>
using namespace std;
int n,k,m=1,father[505];
struct node
{
	int x,y;
	double weight;
}va[250000];

double dis(int x1,int y1,int x2,int y2)
{
	return sqrt((double)((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)));
}
bool cmp(node x,node y)
{
	return x.weight<y.weight;
}

int find(int x)
{
	if(father[x]==x)
		return x;
	else
		return father[x]=find(father[x]);
}

int merge(int x,int y)
{
	int fx=find(x),fy=find(y);
	if(fx==fy)
		return 0;
	else
		father[fx]=fy;
	return 1;
}

void Kruskal()
{
	sort(va+1,va+m+1,cmp);
	for(int i=1;i<=n;i++)
		father[i]=i;
	int num=0;
	for(int i=1;i<=m;i++)
	{
		if(merge(va[i].x,va[i].y))
		{
			num++;
			if(num==n-k)
			{
				cout<<setiosflags(ios::fixed)<<setprecision(2)<<va[i].weight<<endl;
				break;
			}
		}
	}
}
int main()
{
	int a[505],b[505];
	cin>>n>>k;
	if(n<=k)
	{
		cout<<"0.00"<<endl;
		return 0;
	}
	if(k==0)
		k=1;
	for(int i=1;i<=n;i++)
		cin>>a[i]>>b[i];
	for(int i=1;i<=n;i++)
		cin>>a[i]>>b[i];
	for(int i=1;i<=n;i++)
	{
		for(int j=i+1;j<=n;j++)
		{
			va[m].x=i;
			va[m].y=j;
			va[m].weight=dis(a[i],b[i],a[j],b[j]);
			m++;
		}
	}
	m--;
	Kruskal();
	return 0;
}