bzoj 1821: [JSOI2010]Group 部落划分

1821: [JSOI2010]Group 部落划分

Time Limit: 10 Sec   Memory Limit: 64 MB
Submit: 2484   Solved: 1165
[ Submit][ Status][ Discuss]

Description

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

Input

第一行包含两个整数N和K(1< = N < = 1000,1< K < = N)，分别代表了野人居住点的数量和部落的数量。

接下来N行，每行包含两个正整数x,y，描述了一个居住点的坐标(0 < =x, y < =10000)

Output

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

Sample Input

4 2

0 0

0 1

1 1

1 0

Sample Output

1.00

如果两个居住点属于同一个部落，那么就相当于这两个点之间有一条边

直接kruskal，n个点总共n*(n-1)/2条双向边，将这些边按长度从短到长排个序

之后在保证不出现环的情况下一条一条加入图中，加完k条边第k+1条边的长度就是答案

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
typedef struct Res
{
	int x, y;
	double len;
	bool operator < (const Res &b) const
	{
		if(len<b.len)
			return 1;
		return 0;
	}
}Road;
Road s[500005];
int ufs[1005], x[1005], y[1005];
int Find(int x)
{
	if(ufs[x]==0)
		return x;
	return ufs[x] = Find(ufs[x]);
}
int main(void)
{
	int n, k, i, j, m, t1, t2;
	while(scanf("%d%d", &n, &k)!=EOF)
	{
		k = n-k+1;
		for(i=1;i<=n;i++)
			scanf("%d%d", &x[i], &y[i]);
		m = 0;
		memset(ufs, 0, sizeof(ufs));
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=i-1;j++)
			{
				s[++m].x = i, s[m].y = j;
				s[m].len = sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+1.0*(y[i]-y[j])*(y[i]-y[j]));
			}
		}
		sort(s+1, s+m+1);
		for(i=1;i<=m;i++)
		{
			t1 = Find(s[i].x);
			t2 = Find(s[i].y);
			if(t1!=t2)
			{
				ufs[t1] = t2;
				k--;
			}
			if(k==0)
				break;
		}
		printf("%.2f\n", s[i].len);
	}
	return 0;
}