poj 2349(最小生成树应用)

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

思路：由于有S个专门的通道，我们可以先求一次最小生成树，然后对于最小生成树上的边从大到小排序，前S-1条边用S-1个卫星通道连接，那么第S大条边就是我们要找的最小的D了。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<cmath>
 #include<algorithm>
 using namespace std;
 #define MAXN 555
 #define inf 1LL<<60

 struct Node{
     int x,y;
 }node[MAXN];
 double map[MAXN][MAXN];
 double lowcost[MAXN];
 double dist[MAXN];
 bool mark[MAXN];
 int n,m;

 double Cal(int i,int j)
 {
     double d1=1.0*(node[i].x-node[j].x)*(node[i].x-node[j].x);
     double d2=1.0*(node[i].y-node[j].y)*(node[i].y-node[j].y);
     return sqrt(d1+d2);
 }

 int cmp(const double &p,const double &q)
 {
     return p>q;
 }

 double Prim(int u0)
 {
     int cnt=;
     memset(mark,false,sizeof(mark));
     for(int i=;i<=m;i++){
         lowcost[i]=map[u0][i];
     }
     lowcost[u0]=;
     mark[u0]=true;
     for(int i=;i<m;i++){
         double MIN=inf;
         int v=u0;
         for(int j=;j<=m;j++){
             if(!mark[j]&&lowcost[j]<MIN){
                 MIN=lowcost[j],v=j;
             }
         }
         if(MIN==inf)break;
         mark[v]=true;
         dist[cnt++]=lowcost[v];
         for(int k=;k<=m;k++){
             if(!mark[k]&&map[v][k]<lowcost[k]){
                 lowcost[k]=map[v][k];
             }
         }
     }
     sort(dist,dist+cnt,cmp);
     //for(int i=0;i<cnt;i++)printf("%.2lf\n",dist[i]);
     return dist[n-];
 }

 int main()
 {
     int _case,a,b;
     scanf("%d",&_case);
     while(_case--){
         scanf("%d%d",&n,&m);
         for(int i=;i<=m;i++)
             for(int j=;j<=m;j++)
                 map[i][j]=map[j][i]=(i==j?:inf);
         for(int i=;i<=m;i++){
             scanf("%d%d",&node[i].x,&node[i].y);
             for(int j=;j<i;j++){
                 map[i][j]=map[j][i]=Cal(i,j);
             }
         }
         printf("%.2lf\n",Prim());
     }
     return ;
 }