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=0;
    memset(mark,false,sizeof(mark));
    for(int i=1;i<=m;i++){
        lowcost[i]=map[u0][i];
    }
    lowcost[u0]=0;
    mark[u0]=true;
    for(int i=1;i<m;i++){
        double MIN=inf;
        int v=u0;
        for(int j=1;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=1;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
",dist[i]);
    return dist[n-1];
}
    
        
int main()
{
    int _case,a,b;
    scanf("%d",&_case);
    while(_case--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++)
            for(int j=1;j<=m;j++)
                map[i][j]=map[j][i]=(i==j?0:inf);
        for(int i=1;i<=m;i++){
            scanf("%d%d",&node[i].x,&node[i].y);
            for(int j=1;j<i;j++){
                map[i][j]=map[j][i]=Cal(i,j);
            }
        }
        printf("%.2lf
",Prim(1));
    }
    return 0;
}