【最小生成树】P4047 [JSOI2010]部落划分

原创于 2020-08-01 22:14:15 发布 · 140 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

算法同时被 2 个专栏收录

218 篇文章

订阅专栏

最小生成树

4 篇文章

订阅专栏

本文介绍了一种基于最小生成树思想的算法，用于求解如何连接节点以使两个部落间的最小距离达到最大。通过实现该算法，我们首先计算所有节点对之间的距离，然后按距离从小到大排序并逐步构建最小生成树，直到满足部落数量的要求。

由于要求两个部落之间的最小距离最大，可以每次先连接两个最近的节点，就和最小生成树的思路一样了

代码

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int u,v;
    double w;
}a[1000000];
int n,f[10000],cnt,k,l;
int x[10000],y[10000];
void add(int uu,int vv,double ww)  
{a[++cnt].u=uu;a[cnt].v=vv;a[cnt].w=ww;}
bool cmp(node a1,node a2)
{return a1.w<a2.w;}
int find(int u)  
{
    if(f[u]==u) return u;
    else return f[u]=find(f[u]);
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        f[i]=i;
    for(int i=1;i<=n;i++)
        scanf("%d%d",&x[i],&y[i]);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)   
        {
            double s=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
            add(i,j,s);
        }
    sort(a+1,a+cnt+1,cmp);    
    for(int i=1;i<=cnt;i++)    
    {
        int from=find(a[i].u);   
        int to=find(a[i].v);
        if(from==to)
			continue;  
        f[from]=to;              
        l++;                  
        if(l==n-k+1)           
        {
            printf("%.2lf",a[i].w);
            return 0;
        }
    }
    return 0;
}