HDU1875 畅通工程再续【最小生成树】

题意：

在这些小岛中建设最小花费的桥，但是一座桥的距离必须在10 -- 1000之间。

思路：

用最小生成树解决吧，就那两个算法。

代码：

prim

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>

using namespace std;
#define N 105
#define inf 0x3f3f3f3f

int n,i,j,x[N],y[N];
double map[N][N];
bool vis[N];//标记是否已经放入最小生成树的那个集合里了
double low[N];//记录不在已经加入最小生成树的这个集合里的元素到这个 集合的最小距离

double dis(int i,int j)
{
    return sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
}

void init()
{
    for(i=1;i<=n;++i)
        cin>>x[i]>>y[i];
    for(i=1;i<=n;++i)
        for(j=1;j<=n;++j)
        {
            map[i][j]=dis(i,j);
            if(map[i][j]>1000||map[i][j]<10)
            {
                map[i][j]=inf;
            }
        }
}

void prim()
{
    double sum=0;
    memset(vis,0,sizeof(vis));
    int pos=1;//从1开始
    for(i=1;i<=n;++i)//第一次给low赋值
    {
        low[i]=map[1][i];
    }
    vis[1]=1;
    //已经找到一个点1，再找n-1个
    for(i=1;i<n;++i)
    {
        double min=inf;
        for(j=1;j<=n;++j)
        {
            if(!vis[j]&&min>low[j])//找下一个点到这个集合的最小值
            {
                min=low[j];//记下这个最小值
                pos=j;//记下这个点
            }
        }
        if(min==inf)
        {
            cout<<"oh!"<<endl;
            return ;
        }
        sum+=min;
        vis[pos]=1;//把刚刚找到的这个点加入集合
        for(j=1;j<=n;++j) //更新low数组
        {
            if(!vis[j]&&low[j]>map[pos][j])
            {
                low[j]=map[pos][j];
            }
        }
    }
    printf("%.1lf\n",sum*100);
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n;
        init();
        prim();
    }
    return 0;
}

View Code