HDU-1875 畅通工程再续（最小生成树[Prim]）

畅通工程再续

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

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!
  
  

本题除了让自己判断两点之间是否存在边外，依旧是一道裸的最小生成树。只需要注意把两点间距离的平方当作权值以避免精度问题即可。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int INF=0x3f3f3f3f;

int n,u,cnt,vis[105],dis[105],g[105][105];//dis[i]表示点i到最小生成树的距离
double ans;
struct Point {
    int x,y;
}p[105];

void Prim() {
    int i;
    memset(vis,false,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    ans=cnt=dis[1]=0;
    while(true) {
        u=0;
        for(i=1;i<=n;++i)
            if(!vis[i]&&dis[i]<dis[u])
                u=i;
        if(u==0)
            return ;
        vis[u]=true;
        ++cnt;
        ans+=sqrt(double(dis[u]))*100;
        for(i=1;i<=n;++i)
            dis[i]=min(dis[i],g[u][i]);
    }
}

int main() {
    int i,j,T,t,dx,dy;
    scanf("%d",&T);
    while(T--) {
        memset(g,0x3f,sizeof(g));
        scanf("%d",&n);
        for(i=1;i<=n;++i) {
            scanf("%d%d",&p[i].x,&p[i].y);
            for(j=1;j<i;++j) {
                dx=p[i].x-p[j].x,dy=p[i].y-p[j].y;
                t=dx*dx+dy*dy;
                if(100<=t&&t<=1000000)
                    g[i][j]=g[j][i]=t;//为了避免产生精度问题，权值为距离的平方
            }
        }
        Prim();
        if(cnt==n)
            printf("%.1lf\n",ans);
        else
            printf("oh!\n");
    }
    return 0;
}