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;}