POJ 2349 Arctic Network（Prim/Kruskal）

题目链接：
POJ 2349 Arctic Network
题意：
有n个村庄，需要这些村庄实现两两通信（连通）。
有两种通信方式：卫星通信无视距离。无线电通信距离不超过D。
已知可提供p颗卫星，由于这些通信是统一样式的，而且D越大成本越高。为了让成本最小，求出最小的D。
分析：
可提供p颗卫星，每个村庄使用一颗，也就是有p-1条边是不需要考虑的。
又因为将n个点连通，需要最少n-1条边。
所以最终就是要求最小生成树中第（n-1)-(p-1)条边的边长。（把边从小到大排序后）
$CODE$:

//Kruskal Algorithm
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int maxn=510;
const int INF=0x3f3f3f3f;

int T,p,n,tot;
int pre[maxn];
double ans;

struct Point{
    int x,y;
}point[maxn];

struct Edge{
    int u,v;
    double w;
}edge[maxn*maxn];

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

int find(int x)
{
    return pre[x]==x?x:pre[x]=find(pre[x]);
}

double Distance(struct Point a,struct Point b)
{
    int xx=a.x-b.x;
    int yy=a.y-b.y;
    return sqrt(xx*xx+yy*yy);
}

bool cmp(struct Edge a,struct Edge b)
{
    return a.w<b.w;
}

int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d%d",&p,&n);
        for(int i=1;i<=n;i++)
            scanf("%d%d",&point[i].x,&point[i].y);
        tot=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                edge[tot].u=i;
                edge[tot].v=j;
                edge[tot].w=Distance(point[i],point[j]);
                tot++;
            }
        }
        sort(edge,edge+tot,cmp);
        //for(int i=0;i<tot;i++)
        //    printf("edge[%d].u=%d edge[%d].v=%d edge[%d].w=%.2f\n",i,edge[i].u,i,edge[i].v,i,edge[i].w);

        int cnt=0;//cnt记录已经连通边数
        for(int i=0;i<tot;i++)
        {
            int u=edge[i].u;
            int v=edge[i].v;
            int fu=find(u);
            int fv=find(v);
            //printf("u=%d v=%d fu=%d fv=%d\n",u,v,fu,fv);
            //printf("cnt=%d\n",cnt);
            if(fu==fv) continue;
            pre[fu]=fv;
            cnt++;
            if(cnt==n-p)
            {
                ans=edge[i].w;
                break;
            }
        }
        printf("%.2f\n",ans);
    }
    return 0;
}

//Prim Algorithm
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int maxn=510;
const int INF=0x3f3f3f3f;

int T,n,p;
int vis[maxn];
double dis[maxn],map[maxn][maxn];
double ans[maxn];//ans数组记录最小生成树的各条边长

struct Point{
    int x,y;
}point[maxn];

double Distance(struct Point a,struct Point b)
{
    double xx=a.x-b.x;
    double yy=a.y-b.y;
    return sqrt(xx*xx+yy*yy);
}

int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&p,&n);
        for(int i=1;i<=n;i++)
            scanf("%d%d",&point[i].x,&point[i].y);
        for(int i=1;i<=n;i++)
        {
            map[i][i]=0;
            for(int j=i+1;j<=n;j++)
            {
                map[i][j]=map[j][i]=Distance(point[i],point[j]);
            }
        }
/*
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
                printf("%.2f ",map[i][j]);
            printf("\n");
        }
*/
        memset(vis,0,sizeof(vis));
        vis[1]=1;//选取1作为开始点
        for(int i=1;i<=n;i++)
        {
             dis[i]=map[1][i];
             //printf("dis[%d]=%.2f ",i,dis[i]);
        }
        //printf("\n");

        int tot=0;

        for(int i=1;i<n;i++)
        {//还需要连接n-1个点
            int k=-1;
            double tmp=INF;
            for(int j=1;j<=n;j++)
            {
                if(!vis[j]&&dis[j]<tmp)
                {
                    tmp=dis[j];
                    k=j;
                }
            }
            vis[k]=1;
            ans[tot++]=tmp;
            //printf("k=%d tmp=%.2f\n",k,tmp);
            for(int j=1;j<=n;j++)
            {
                if(!vis[j]&&dis[j]>map[k][j]) dis[j]=map[k][j];
                //printf("dis[%d]=%.2f ",j,dis[j]);
            }
            //printf("\n");
        }
        sort(ans,ans+tot);
        printf("%.2f\n",ans[n-p-1]);//因为边下标是从0开始的所以第n-p条边的下标是n-p-1
    }
    return 0;
}