poj2349Arctic Network

题目链接：http://poj.org/problem?id=2349

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=550;
int n,m;
int f[maxn];
double ans[maxn];
struct node
{
    int x,y;
}p[maxn];
struct edge
{
    int u,v;
    double w;
    bool operator<(const edge& a)
    {
        return w<a.w;
    }
}e[maxn*maxn];

void init()
{
    for(int i=0;i<=n;i++)
        f[i]=i;
}

int gf(int x)
{
    return x==f[x]?x:f[x]=gf(f[x]);
}

void uni(int a,int b)
{
    int pa=gf(a);
    int pb=gf(b);
    f[pb]=pa;
}
double getdis(int i,int j)
{
    double dx=p[i].x-p[j].x;
    double dy=p[i].y-p[j].y;
    return sqrt(dx*dx+dy*dy);
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&m,&n);
         init();
        for(int i=0;i<n;i++)
            scanf("%d%d",&p[i].x,&p[i].y);
        int cnt=0,cas=0;
        for(int i=0;i<n;i++)
            for(int j=i+1;j<n;j++)
        {
            e[cnt].u=i;
            e[cnt].v=j;
            e[cnt].w=getdis(i,j);
            cnt++;
        }
        if(m>=n) {printf("0.00
");continue;}
        sort(e,e+cnt);
        for(int i=0;i<cnt;i++)
        {
            int v=e[i].v;
            int u=e[i].u;
            if(gf(u)!=gf(v))
                {
                uni(u,v);
                cas++;
                ans[cas]=e[i].w;
                if(cas==n-1) break;
                }
        }
        printf("%.2f
",ans[n-m]);


    }
}