【bzoj1821】[JSOI2010]Group 部落划分 Group Kruskal

题目描述

聪聪研究发现，荒岛野人总是过着群居的生活，但是，并不是整个荒岛上的所有野人都属于同一个部落，野人们总是拉帮结派形成属于自己的部落，不同的部落之间则经常发生争斗。只是，这一切都成为谜团了——聪聪根本就不知道部落究竟是如何分布的。 不过好消息是，聪聪得到了一份荒岛的地图。地图上标注了N个野人居住的地点（可以看作是平面上的坐标）。我们知道，同一个部落的野人总是生活在附近。我们把两个部落的距离，定义为部落中距离最近的那两个居住点的距离。聪聪还获得了一个有意义的信息——这些野人总共被分为了K个部落！这真是个好消息。聪聪希望从这些信息里挖掘出所有部落的详细信息。他正在尝试这样一种算法： 对于任意一种部落划分的方法，都能够求出两个部落之间的距离，聪聪希望求出一种部落划分的方法，使靠得最近的两个部落尽可能远离。 例如，下面的左图表示了一个好的划分，而右图则不是。请你编程帮助聪聪解决这个难题。


输入

第一行包含两个整数N和K(1< = N < = 1000,1< K < = N)，分别代表了野人居住点的数量和部落的数量。
接下来N行，每行包含两个正整数x,y，描述了一个居住点的坐标(0 < =x, y < =10000)
输出

输出一行，为最优划分时，最近的两个部落的距离，精确到小数点后两位。

样例输入

4 2

0 0

0 1

1 1

1 0

样例输出

1.00

题解

“最小生成森林”

题目要求保留k个连通块，就是加n-k条边。于是按照Kruskal的做法查找到第n-k+1条边时输出即可。

#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
struct data
{
int x , y , z;
}a[1000010];
int dx[1010] , dy[1010] , cnt , f[1010];
bool cmp(data a , data b)
{
return a.z < b.z;
}
int find(int x)
{
return x == f[x] ? x : f[x] = find(f[x]);
}
int main()
{
int n , k , i , j , cedge = 0 , tx , ty;
scanf("%d%d" , &n , &k);
for(i = 1 ; i <= n ; i ++ )
scanf("%d%d" , &dx[i] , &dy[i]);
for(i = 1 ; i <= n ; i ++ )
for(j = i + 1 ; j <= n ; j ++ )
a[++cnt].x = i , a[cnt].y = j , a[cnt].z = (dx[i] - dx[j]) * (dx[i] - dx[j]) + (dy[i] - dy[j]) * (dy[i] - dy[j]);
sort(a + 1 , a + cnt + 1 , cmp);
for(i = 1 ; i <= n ; i ++ )
f[i] = i;
for(i = 1 ; i <= cnt ; i ++ )
{
tx = find(a[i].x) , ty = find(a[i].y);
if(tx != ty)
{
if(cedge == n - k)
{
printf("%.2lf\n" , sqrt(a[i].z));
return 0;
}
else f[tx] = ty , cedge ++ ;
}
}
return 0;
}