畅通工程再续 prim 1

最新推荐文章于 2020-02-21 20:24:10 发布

心夏心冬

最新推荐文章于 2020-02-21 20:24:10 发布

阅读量309

点赞数

分类专栏：最小生成树

本文链接：https://blog.csdn.net/xinxiaxindong/article/details/60329492

版权

最小生成树专栏收录该内容

4 篇文章 0 订阅

订阅专栏

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组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<cmath>
#include<cstring>
#define INF 0x3f3f3f3f
double map[1020][1020];
bool v[1020];
double d[1020];
int n;
int flag = 1;

double prim(){
    for (int i = 1; i <= n; ++i){
        d[i] = 2047.0;
    }
    memset(v,false,sizeof(v));
    int pt;
    double res,min;
    v[1] = true; pt = 1;
    while (true){
        for (int i = 1; i <= n; ++i){
            if (!v[i] && map[pt][i] && d[i] > map[pt][i])
                d[i] = map[pt][i];
        }
        min = INF,pt = -1;
        for (int i = 1; i <= n; ++i){
            if (!v[i] && min > d[i]){
                min = d[i];
                pt = i;
            }
        }
        if (min == 2047) flag = 0;
        if (pt == -1 || min == 2047) break;
        v[pt] = true;
        res += min;
    }
    return res;
}
struct node {
    int x,y;
}land[1020];

double dis(node a,node b){
    double d = sqrt(1.0*(a.x - b.x)*(a.x - b.x) + 1.0*(a.y - b.y)*(a.y - b.y));
    return d;
}
int main(){
    int t;
    scanf("%d",&t);
    while (t--){
        scanf("%d",&n);
        for (int i = 1; i <= n; ++i){
            scanf("%d%d",&land[i].x,&land[i].y);
        }
        for (int i = 1; i <= n; ++i){
            for (int j = 1; j <= n; ++j){
                if (i == j)
                   map[j][i] = map[i][j] = 0;
                else if(dis(land[i],land[j]) >= 10.0 && dis(land[i],land[j]) <= 1000.0)
                    map[i][j] = dis(land[i],land[j]);
                else
                    map[j][i] =map[i][j] = 0 ;
            }
        }

        double t = prim();
        if (flag == 1) printf("%.1lf\n",t*100.0);
        else printf("oh!\n");
    }
    return 0;
}