【洛谷 1991】 无线通讯网

跑最短路就行 最后留上几条长边

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#define MAXN 1000000
struct EDGE
{
    int x,y;
    double L;
}a[MAXN];
int cnt;
int t[600][2];
int s,p;
int fa[MAXN];
int Find(int x)
{
    if(x!=fa[x]) fa[x]=Find(fa[x]);
    return fa[x];
}
double D(int x,int y)
{
//  std::cout<<x<<' '<<y<<' ';
    int t1=t[x][0]-t[y][0],t2=t[x][1]-t[y][1];// std::cout<<sqrt(t1*t1+t2*t2)<<std::endl;
    return sqrt(t1*t1+t2*t2);
}
bool cmp(EDGE a,EDGE b){return a.L<b.L;}
int main()
{
    scanf("%d %d",&s,&p);
    for(int i=1;i<=p;i++) scanf("%d %d",&t[i][0],&t[i][1]);
    for(int i=1;i<=p;i++)
        for(int j=i+1;j<=p;j++)
            a[++cnt].x=i,a[cnt].y=j,a[cnt].L=D(i,j); 
    for(int i=1;i<=cnt;i++) fa[i]=i;
    std::sort(a+1,a+cnt+1,cmp);
    s=p-s;
    double ans=0;
    for(int i=1;i<=cnt&&s!=0;i++)
        if(Find(a[i].x)!=Find(a[i].y))
        {
            fa[Find(a[i].x)]=Find(a[i].y);
            ans=a[i].L;
            s--;
        }
    printf("%.2lf\n",ans);
    return 0;
}