Constructing Roads 最小生成树 消除已经建过的路的影响

题目中两个村庄如果没有路的话长度为0,为了消除已经建过的路对结果的影响,将没有路的长度设置为-1,已经建过路的村庄之间的路长度为0
然后用prime算法求出最小生成树即可

#include <cstring>
#include <iostream>
#define mem(a, v) memset(a, v, sizeof(a))
#define fre(f) freopen(f ".in", "r", stdin), freopen(f ".out", "w", stdout)
#define inf 0x3f3f3f3f
const int N = 1e2 + 5;
using namespace std;

int g[N][N], vis[N], sum = 0;

int main()
{
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    mem(vis, 0);
    int n, m, x, y;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            cin >> g[i][j];
            if (g[i][j] == 0) g[i][j] = -1; //没有路,长度为-1
        }
    }
    cin >> m;
    for (int i = 1; i <= m; i++)
    {
        cin >> x >> y;
        g[x][y] = g[y][x] = 0; //建过路了就将长度设置为0,加入了树中对结果也不会有影响
    }
    vis[1] = 1;
    for (int i = 1; i < n; i++)
    {
        int minlen = inf, ans;
        for (int j = 1; j <= n; j++)
        {
            if (!vis[j]) continue;
            for (int k = 1; k <= n; k++)
            {
                if (vis[k] || k == j || g[j][k] == -1) continue;
                if (g[j][k] < minlen)
                {
                    minlen = g[j][k];
                    ans = k;
                }
            }
        }
        vis[ans] = 1;
        sum += minlen;
    }
    cout << sum;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值