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

   
   

    
    2
2
10 10
20 20
3
1 1
2 2
1000 1000
   
   

 

Sample Output

   
   

    
    1414.2
oh!
   
   
   
   


   
   

代码（Kruskal算法）：

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;

#define maxn 100+10
#define maxv 10000+10
struct 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+10
const 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;
}