部落划分 最小生成树+并查集+贪心

n n n个野人居住坐标,野人总共被分为了 k k k个部落,两个部落的距离,定义为部落中距离最近的那两个居住点的距离。对于任意一种部落划分的方法,都能够求出两个部落之间的距离,聪聪希望求出一种部落划分的方法,使靠得最近的两个部落尽可能远离。
一道最小生成树的变式题,把所有野人之间的距离预处理一下,每次找到最小的一条边,把这两个野人合成一个部落,一直重复,一共有k个部落,我们只需要找n-k条边,这些都是最小距离,所以剩下的就是最大了。

#include
using namespace std;
struct node
{
	int u,v;
	double w;
}a[1000000];
int x[10000],y[10000],f[1000000];
int n,K,num=0;
void add(int u,int v,double w)
{
	a[++num].u=u;
	a[num].v=v;
	a[num].w=w;
}
bool cmp(node X,node Y)
{
	return X.w<Y.w;
}
int find(int k)
{
	if(f[k]==k)return k;
	else return f[k]=find(f[k]);
}
double dis(double x1,double x2,double y1,double y2)
{
	return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}
void kus()
{
	int l=0;
	sort(a+1,a+num+1,cmp);
	for(int i=1;i<=num;i++)
	{
		int U=find(a[i].u);
		int V=find(a[i].v);
		if(U==V)continue;
		f[U]=V;l++;
		if(l==n-K+1)          
        {
            printf("%.2lf",a[i].w);
            return ;
        }
	}
    return ;
}
int main()
{
	cin>>n>>K;
	for(int i=1;i<=n;i++)
	{
		f[i]=i;
		cin>>x[i]>>y[i];
	}
	for(int i=1;i<=n;i++)
	for(int j=i+1;j<=n;j++)
	{
		double ans=dis(x[i],x[j],y[i],y[j]);
		add(i,j,ans);
	}
	kus();
	return 0;
}

你可能感兴趣的:(算法,数据结构,图论)