poj2349 (kruskal算法)

//求最小生成樹的第k長邊
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
int bin[550];
struct op
{
    int a,b;
double c;
}s[250001];
int cmp(const void*a,const void*b)
{
   op*aa=(op*)a;
   op*bb=(op*)b;
   return aa->c>bb->c?1:-1;
}
int find(int x)
{
    while(bin[x]!=x)
        x=bin[x];
    return x;
}
bool merge(int a,int b)
{
if(find(a)!=find(b))
{
bin[find(b)]=find(a);return 1;
}
else return 0;
}
int main()
{
    int i,j,t,k,n,m;
int x[510],y[510];
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&k,&n);m=0;
for(i=1;i<=n;i++)
{
bin[i]=i;
  scanf("%d%d",&x[i],&y[i]);
}
for(i=1;i<=n;i++)
{
for(j=i+1;j<=n;j++)
{
              s[m].a=i;s[m].b=j;
 s[m].c=sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+1.0*(y[i]-y[j])*(y[i]-y[j]));
 m++;
}
}
        qsort(s,m,sizeof(s[0]),cmp);int sum=0;
for(i=0;i<m;i++)
{
if(merge(s[i].a,s[i].b))
            {
 sum++;
 if(sum==n-k){printf("%.2f\n",s[i].c);break;}
}
}
}
     return 0;
}
 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章