E - 百岛湖的故事(最小生成树Prim算法)

相信大家都听说一个“百岛湖”的地方吧,百岛湖的居民生活在不同的小岛中,当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖,发展首先要解决的问题当然是交通问题,政府决定实现百岛湖的全畅通!经过考察小组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!
题解
这是一个最小生成树的普及题目,今天理解了Prim算法的思路,题目自然而然就过了。
先来讲讲 Prim算法
假设我们有n个点:
cost[i][j] :点i和点j之间的距离。
mincost[i]:点i到树的最短距离。
V:树外所有的点。
第一步:
我们从V里选一个点x 加入树里面,此时V到树的距离就是V到x的距离,我们就把mincost设置成V到x的距离。
第二步:
找出V里离树最近的点x,将x加入树。
第三步:
因为树多了一个点x,而此时mincost表示的是x没加之前,V到树的最短距离。
所以我们需要更新一下mincost:mincost[ i ]=min( mincost[ i ],cost [ v ][ i ] )
然后一直重复第二步和第三步,直到所有的点都加入树就行了。
不过这题有个条件限制,不是所有两点之间都能够连接,所以如果V里面的点都无法加入树,那么就结束。
AC代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 110;
const int times = 100;
const int INF = (unsigned)(-1) >> 1;
int n;
struct Pos
{
    int x, y;
};
Pos island[maxn];          //小岛的位置
double cost[maxn][maxn];   //两个小岛之间的最短距离
double mincost[maxn];      //树外的点到树的最小距离,这个可以用堆来维护
bool uesd[maxn];           //小岛是否加入了树
double setup(Pos a, Pos b) //计算两个小岛之间搭桥的花费
{
    double x = a.x - b.x, y = a.y - b.y;
    double dist = sqrt(x * x + y * y);
    if (dist >= 10 * times && dist <= 1000 * times)
        return dist;
    return INF;
}
double solove()
{
    for (int i = 0; i < n; i++)
    {
        mincost[i] = INF;
        uesd[i] = false;
    }
    mincost[0] = 0;
    double res = 0;
    while (true)
    {
        int v = -1;
        for (int i = 0; i < n; i++)
        {
            if (!uesd[i] && (v == -1 || mincost[i] < mincost[v]))
                v = i;
        }
        if (mincost[v] == INF)
            return -1;
        if (v == -1)
            return res;
        uesd[v] = true;
        res += mincost[v];
        for (int i = 0; i < n; i++)//更新树外的点到树的最短距离
            mincost[i] = min(mincost[i], cost[v][i]);
    }
}
int main(void)
{
    int t;
    cin >> t;
    while (t--)
    {
        cin >> n;
        for (int i = 0; i < n; i++)
        {
            cin >> island[i].x >> island[i].y;
            island[i].x *= times, island[i].y *= times;
        }
        for (int i = 0; i < n - 1; i++)
            for (int j = i + 1; j < n; j++)
                cost[j][i] = cost[i][j] = setup(island[i], island[j]); //计算每两个小岛之间搭桥的花费
        double ans = solove();
        if (ans == -1)
            cout << "oh!" << endl;
        else
            printf("%.1lf\n", ans);
    }
    return 0;
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值