BZOJ1821部落划分

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

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

struct edge{
    int s,t;
    double v;
}e[1000001];
int tot=0;
int x[1001],y[1001],fa[1001]; 

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

void work(int a,int b)
{
    double now;
    now=sqrt((x[a]-x[b])*(x[a]-x[b])+(y[a]-y[b])*(y[a]-y[b]));
    tot++;
    e[tot].s=a; e[tot].t=b; e[tot].v=now;
}

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

int main()
{
    int n,k;
    cin >> n >> k;
    for (int i=1; i<=n; i++) cin >> x[i] >> y[i];
    for (int i=1; i<=n; i++)
        for (int j=1; j<=n; j++)
            work(i,j);
     sort(e+1,e+tot+1,cmp); 
     for (int i=1; i<=n; i++) fa[i]=i;
     for (int i=1; i<=tot; i++)
         if (find(e[i].s)!=find(e[i].t))
             if (n>k){ n--;    fa[find(e[i].s)]=fa[e[i].t]; }
                 else { printf("%.2lf",e[i].v); break; }
     return 0;
 }