P4047 [JSOI2010]部落划分(kruscal)

130 阅读3分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第30天,点击查看活动详情(juejin.cn/post/709970… "juejin.cn/post/709970…")

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

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

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

例如,下面的左图表示了一个好的划分,而右图则不是。请你编程帮助聪聪解决这个难题。 image.png 分析:这是一个划分区间问题,那么也比较容易想到用kruscal求最小生成树问题,因为kruscal求最小生成树的思想的本质就是贪心,优先加入不在同一集合且距离较小的边,所以我们按照这样的思路进行集合合并,当集合合并至只剩下k个时,下一个能够将不同集合合并的边就是我们要求的距离,因为我们保证了边权最小的边都在一个集合内部,把边权较大的边留在了集合与集合之间,所以我们下一个可以用来合并两个不同集合的边就是我们的答案。

下面是代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
using namespace std;
const int N=2e6+100;
int x[N],y[N],fu[N];
struct node{
	int u,v;
	double w;
}p[N];
bool cmp(node a,node b)
{
	return a.w<b.w;
}
double dis(int i,int j)
{
	return sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
}
int find(int x)
{
	if(x!=fu[x]) fu[x]=find(fu[x]);
	return fu[x];
}
int main()
{
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n;i++)
		scanf("%d%d",&x[i],&y[i]),fu[i]=i;
	int cnt=0;
	for(int i=1;i<=n;i++)
	for(int j=i+1;j<=n;j++)
		p[++cnt]={i,j,dis(i,j)};
	sort(p+1,p+cnt+1,cmp);
	int ans=n;//ans为当前部落数量 
	for(int i=1;i<=cnt;i++)
	{
		int f1=find(p[i].u),f2=find(p[i].v);
		if(f1==f2) continue;
		fu[f1]=f2;
		ans--;
		if(ans==k-1)
		{
			printf("%.2lf\n",p[i].w);
			break;
		}
	}
	return 0;
}