hdu 畅通工程再续（最小生成树）（Prim算法 && Kruskal算法）

畅通工程再续

Problem Description

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组RPRush对百岛湖的情况充分了解后，决定在符合条件的小岛间建上桥，所谓符合条件，就是2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。

 

Input

输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数，接下来是C组坐标，代表每个小岛的坐标，这些坐标都是 0 <= x, y <= 1000的整数。

 

Output

每组输入数据输出一行，代表建桥的最小花费，结果保留一位小数。如果无法实现工程以达到全部畅通，输出”oh!”.

 

Sample Input

2210 1020 2031 12 21000 1000

 

Sample Output

1414.2oh!

代码（Kruskal算法）：

#include<stdio.h>#include<math.h>#include<string.h>#include<algorithm>using namespace std;#define maxn 100+10#define maxv 10000+10struct edge{    int u,v;    double w;} e[maxv];int pre[maxn],x[maxn],y[maxn];int n,len;bool cmp(edge a,edge b){    return a.w<b.w;}void init(){    for(int i=1; i<maxn; i++)        pre[i]=i;}int Find(int x){    if(x==pre[x])        return x;    else    {        pre[x]=Find(pre[x]);        return pre[x];    }}int join(int x,int y){    int fx=Find(x),fy=Find(y);    if(fx!=fy)    {        pre[fy]=fx;        return 1;    }    return 0;}void Kruskal(){    sort(e,e+len,cmp);    int cont=0;    double sum=0;    for(int i=1; i<len; i++)    {        if(e[i].w<10||e[i].w>1000)            continue;        if(join(e[i].u,e[i].v))        {            cont++;            sum+=e[i].w*100;        }        if(cont==n-1)            break;    }    if(cont<n-1)        printf("oh!\n");    else        printf("%.1lf\n",sum);}int main(){    int t;    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        init();        int i,j;        double w;        len=1;        for(i=1; i<=n; i++)        {            scanf("%d%d",&x[i],&y[i]);            for(j=1; j<i; j++)            {                w=sqrt((double)(x[i]-x[j])*(double)(x[i]-x[j])+(double)(y[i]-y[j])*(double)(y[i]-y[j]));                e[len].u=i,e[len].v=j,e[len++].w=w;            }        }        Kruskal();    }    return 0;}

代码（Prim算法）：

#include<stdio.h>#include<math.h>#include<string.h>#define maxn 100+10const double inf=0x3f3f3f3f;double e[maxn][maxn],d[maxn];bool vis[maxn];int x[maxn],y[maxn];int n;void Prim(){    int i,j,v;    double minn;    for(i=1; i<=n; i++)    {        d[i]=e[1][i];        vis[i]=0;    }    vis[1]=1;//已经找到了一个点    double sum=0;    for(j=1; j<n; j++)//还需找n-1个点    {        minn=inf;        for(i=1; i<=n; i++)            if(!vis[i]&&d[i]<minn)                minn=d[i],v=i;        if(minn==inf)        {            printf("oh!\n");            return ;        }        vis[v]=1,sum+=d[v];        for(i=1; i<=n; i++)            if(!vis[i]&&d[i]>e[v][i])                d[i]=e[v][i];    }    printf("%.1lf\n",sum*100);}int main(){    int t;    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        int i,j;        double w;        for(i=1; i<=n; i++)            for(j=1; j<=n; j++)                i==j?e[i][j]=0:e[i][j]=e[j][i]=inf;        for(i=1; i<=n; i++)        {            scanf("%d%d",&x[i],&y[i]);            for(j=1; j<i; j++)            {                w=sqrt((double)(x[i]-x[j])*(double)(x[i]-x[j])+(double)(y[i]-y[j])*(double)(y[i]-y[j]));                if(w>1000||w<10)                    e[i][j]=e[j][i]=inf;                else                    e[i][j]=e[j][i]=w;            }        }        Prim();    }    return 0;}