【BZOJ 1821】[JSOI2010]Group 部落划分 Group(并查集)

[BZOJ 1821][JSOI2010]Group 部落划分 Group(并查集)

题面

题意：

聪聪研究发现，荒岛野人总是过着群居的生活，但是，并不是整个荒岛上的所有野人都属于同一个部落，野人们总是拉帮结派形成属于自己的部落，不同的部落之间则经常发生争斗。只是，这一切都成为谜团了——聪聪根本就不知道部落究竟是如何分布的。 不过好消息是，聪聪得到了一份荒岛的地图。地图上标注了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

HINT
Source
JSOI2010第二轮Contest1

IDEA：

先$O(n^2)$的建好边，两点之间边的权值即为两点的欧几里得距离，然后按从小到大排序，直到分成$k$块。

code:

/**************************************************************
    Problem: 1821
    User: codancer
    Language: C++
    Result: Accepted
    Time:664 ms
    Memory:20224 kb
****************************************************************/
 
#include<bits/stdc++.h>
 
using namespace std;
const int N = 1e3+100;
int par[N],ran[N];
int n,k;
map<int,pair<double,double> > point;
double dis(int a,int b){
    return sqrt((point[a].first-point[b].first)*(point[a].first-point[b].first)+(point[a].second-point[b].second)*(point[a].second-point[b].second));
}
void init(){
    for(int i=1;i<=n;i++) par[i]=i,ran[i]=1;
}
int find(int x){
    return x==par[x]?par[x]:find(par[x]);
}
void unite(int x,int y){
    x=find(x);
    y=find(y);
    if(x==y) return ;
    if(ran[x]>ran[y]){
        par[x]=y;
    }
    else{
        par[y]=x;
        if(ran[x]==ran[y]) ran[x]++;
    }
}
bool same(int a,int b){
    return find(a)==find(b);
}
struct edge{
    int u,v;
    double cost;
}e[N*N];
bool cmp(edge a,edge b){
    return a.cost<b.cost;
}
int main(){
    double x,y;
    cin>>n>>k;
    init();
    for(int i=1;i<=n;i++) cin>>x>>y,point[i].first=x,point[i].second=y;
    int cnt=0;
    for(int i=1;i<=n;i++){
        for(int j=i+1;j<=n;j++){
            e[++cnt].u=i;e[cnt].v=j;e[cnt].cost=dis(i,j);
        }   
    }
    sort(e+1,e+cnt+1,cmp);
    // for(int i=1;i<=cnt;i++) cout<<e[i].u<<" "<<e[i].v<<" "<<e[i].cost<<endl;
    double ans=0;
    for(int i=1;i<=cnt;i++){
        if(!same(e[i].u,e[i].v)){
            if(n>k){
                n--;
                unite(e[i].u,e[i].v);
            }
            else{
                ans=e[i].cost;break;
            }
        }
    }
    printf("%.2lf\n",ans);
    return 0;
}