最小生成树-洛谷P4047[JSOI2010]部落划分

题目描述
聪聪研究发现，荒岛野人总是过着群居的生活，但是，并不是整个荒岛上的所有野人都属于同一个部落，野人们总是拉帮结派形成属于自己的部落，不同的部落之间则经常发生争斗。只是，这一切都成为谜团了——聪聪根本就不知道部落究竟是如何分布的。

不过好消息是，聪聪得到了一份荒岛的地图。地图上标注了 nn 个野人居住的地点（可以看作是平面上的坐标）。我们知道，同一个部落的野人总是生活在附近。我们把两个部落的距离，定义为部落中距离最近的那两个居住点的距离。聪聪还获得了一个有意义的信息——这些野人总共被分为了 kk 个部落！这真是个好消息。聪聪希望从这些信息里挖掘出所有部落的详细信息。他正在尝试这样一种算法：

对于任意一种部落划分的方法，都能够求出两个部落之间的距离，聪聪希望求出一种部落划分的方法，使靠得最近的两个部落尽可能远离。

例如，下面的左图表示了一个好的划分，而右图则不是。请你编程帮助聪聪解决这个难题。

输入格式
输入文件第一行包含两个整数 nn 和 kk，分别代表了野人居住点的数量和部落的数量。

接下来 nn 行，每行包含两个整数 xx，yy，描述了一个居住点的坐标。

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

输入输出样例
输入 #1
4 2
0 0
0 1
1 1
1 0
输出
1.00
输入
9 3
2 2
2 3
3 2
3 3
3 5
3 6
4 6
6 2
6 3
输出
2.00

题解：将每个点的距离转换成kru需要的点到点的距离。然后套kru模板就好了

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;

const int N = 1000005;

int n,m;
int f[N];
int k;
int cnt;
int u,v;
int flag;

struct node{
	int x,y;
	double v;
}q[N],p[N];

double cal(int x, int y){
	return sqrt(pow((p[y].y - p[x].y),2) + pow((p[y].x - p[x].x), 2));
}

int find(int x){
	return f[x] == x ? x : f[x] = find(f[x]);
}

void kru(){
	int num = 0;
	for(int i = 1; i <= k; i++){
		if(num == n - m) flag = 1;
		u = find(q[i].x), v= find(q[i].y);
		if(u != v){
			f[u] = v;
			num++;
			if(flag == 1){
				printf("%.2lf", q[i].v);
				return;
			}
		}
		
	}
}

bool cmp(node a, node b){
	return a.v < b.v;
}

int main(){
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++){
		scanf("%d%d", &p[i].x, &p[i].y);
	}
	for(int i = 1; i <= n; i++){
		for(int j = 1; j < i; j++){
			q[++k].x = i, q[k].y = j, q[k].v = cal(i,j);
		}
	}
	sort( q + 1, q + 1 + k, cmp);
	for(int i = 1; i <= n; i++) f[i] = i;
	kru();
	return 0;
}