POJ2349---Arctic Network

求最小生成树第s大的数。

#include<stdio.h>
#include<string.h>
#include<set>
#include<math.h>
#include<algorithm>
using namespace std;
#define ll long long
#define inf 0xfffff
double px[500];
int p[510],n,m,s,flag,ans;
int x[510],y[510];
struct e
{
    int a,b;
    double c;
}path[250000];
int cmp(e a,e b)
{
    return a.c<b.c;
}
int find(int k)
{
    while(p[k]>0)
    {
        k=p[k];
    }
    return k;
}
double dist(int i,int j)
{
    double dist=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
    return dist;
}
void kruskal(int a,int b,double c)
{
    int f1=find(a);
    int f2=find(b);
    if(f1!=f2)
    {
        p[f1]=f2;
        px[ans++]=c;
     //   printf("c=%f\n",c);
    }
}
int cmp2(double a,double b)
{
    return a>b;
}
void inti()
{
    for(int i=1;i<=n;i++)
        p[i]=0;
}
int main()
{   int tt;
    scanf("%d",&tt);
    while(tt--)
    {scanf("%d%d",&s,&n);
        int num=1;
        inti();
        ans=0;
        int aa,bb,cc;
        for(int i=1;i<=n;i++)
        {   scanf("%d%d",&x[i],&y[i]);
            for(int j=1;j<=i;j++)
            if(j<i){
                path[num].a=i;
                path[num].b=j;
                path[num].c=dist(i,j);
                num++;
      //          printf("%d %d %f\n",i,j,dist(i,j));
               }
        }
    //    printf("num=%d\n",num);
        sort(path+1,1+path+num,cmp);
    //    for(int i=1;i<=num;i++)
   //     printf("%d %d %f\n",path[i].a,path[i].b,path[i].c);
        for(int i=1;i<=num;i++)
        kruskal(path[i].a,path[i].b,path[i].c);
        sort(px,px+ans,cmp2);
//        for(int i=ans-1;i>=0;i--)
//            printf("  %f ",px[i]);
//        printf("\nans=%d  %f\n",ans,px[0]);
         printf("%.2f\n",px[s-1]);
    }
}