POJ 1797 最短路问题迪杰斯特拉变形

迪杰斯特拉算法变形求最大载重路径

原创于 2020-03-07 15:16:40 发布 · 5.2k 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种迪杰斯特拉算法的变形应用，用于解决寻找两个城市间能承载最大载重的道路问题。通过调整松弛操作和初始化参数，算法能够在给定城市和道路承重条件下，计算出从起点到终点的最大载重能力。

在这里插入图片描述
题目大意，给你n个城市，m条道路，每条道路有最大承重量。问你要从城市1到城市n能接受的最大载重是多少。

是迪杰斯特拉算法变形，将松弛操作改成d[i]=max(d[i],min(d[v],mp[i][v])。还有相应的初始化的0和INF修改一下即可。

int d[MAX], t[MAX], mp[MAX][MAX];
int main()
{
    int n, m, time;
    cin >> time;
    for (int kkk = 1; kkk <= time; kkk++)
    {
        scanf("%d%d", &n, &m);
        fill(d, d + MAX, -1);
        fill(t, t + MAX, 0);
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
                if (i == j)
                    mp[i][j] = INF;
                else
                    mp[i][j] = -1;
        while (m--)
        {
            int a, b, c;
            scanf("%d%d%d", &a, &b, &c);
            if (mp[a][b] < c)
                mp[a][b] = mp[b][a] = c;
        }
        d[1] = INF;
        for (int l = 1; l <= n; l++)
        {
            int v;
            int minn = -1;
            for (int i = 1; i <= n; i++)
                if (!t[i] && d[i] > minn)
                {
                    minn = d[i];
                    v = i;
                }
            t[v] = 1;
            for (int i = 1; i <= n; i++)
                d[i] = max(d[i], min(d[v], mp[v][i]));
        }
        printf("Scenario #%d:\n%d\n\n", kkk, d[n]);
    }
    return 0;
}