hdu 1879 畅想工程再续（prim算法）

问题描述

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

 

输入

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

 

输出

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

 

样例输入

     
     
     
     

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

 

样例输出

   
   
   
   

    
    
    
     1414.2
oh!
        
   
   
   
   

首先将图存下来，先存到一个结构体中，再存到ma[i][j]数组当中，需要注意的是，当不满足条件时，设距离为无穷大。剩下的就是裸prim()

代码如下：

#include <iostream>
#include <cmath>
#include <iomanip>
using namespace std;

const int INF=99999999;
double dis[105];double ma[105][105];int vis[105];
int n;
struct node
{
    double x,y;
}no[105];

double distence(int a,int b)
{
    return sqrt(pow(no[a].x-no[b].x,2)+pow(no[a].y-no[b].y,2));
}

void build_map()
{

    for(int i=1;i<=n;i++)
        for(int j=i;j<=n;j++)
        {

            double temp=distence(i,j);
            if(temp>=10&&temp<=1000)
            {
                ma[i][j]=temp;
                ma[j][i]=temp;
            }

            else
            {
                 ma[i][j]=INF;
                 ma[j][i]=INF;
            }

        }
}

double prim()
{
    for(int i=1;i<=n;i++)
    {
        vis[i]=0;
        dis[i]=ma[i][1];
    }
   vis[1]=1;
    double sum=0;
    for(int i=1;i<n;i++)//选n-1条边
    {
        double minn=INF+1;int temp;
        for(int j=1;j<=n;j++)
            if(!vis[j]&&dis[j]<minn)
            {
                temp=j;
                minn=dis[j];
            }
            vis[temp]=1;
            if(minn==INF)return -1;
            sum+=minn;
            for(int j=1;j<=n;j++)
                if(!vis[j]&&ma[temp][j]<dis[j])
                    dis[j]=ma[temp][j];
    }
    return sum;

}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n;
        int x,y;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                ma[i][j]=INF;
        for(int i=1;i<=n;i++)
            cin>>no[i].x>>no[i].y;
        build_map();
       double ans=prim();
       if(ans==-1)
        cout<<"oh!"<<endl;
       else
        cout<<fixed<<setprecision(1)<<ans*100<<endl;
    }
}