Floyd算法求最短路

S_keo

于 2022-11-15 23:43:42 发布

阅读量410

点赞数

分类专栏：算法基础课笔记文章标签：算法

本文链接：https://blog.csdn.net/S_keo/article/details/127876354

版权

算法基础课笔记专栏收录该内容

8 篇文章 1 订阅

订阅专栏

Floyd算法（基于动态规划）：用于求多源汇最短路

初始化：用邻接矩阵d[i,j]存储中所有的边，floyd算法就是三重循环

for(k = 1;k <= n;k ++){

for(i = 1;i <= n;i ++){

for(j = 1;j <= n;j ++){

d(i,j) = min(d(i,j) , d(i,k) + d(k,j))//更新

}

}

}

结束之后d[i,j]存的就是从i到j的最短路的长度。

floyd算法本质上是基于动态规划，它的状态表示是三维的d[k,i,j]，表示从i点只经过1~k这些中间点到达j的最短距离。更新时先枚举k，因为k是阶段，算k时要从k-1转移过来。

即 $d[k,i,j] = d[k-1,i,k]+d[k-1,k,j]$ ，

由于第一维没什么用，可以被优化掉，所以最终表示为 $d[i,j] = d[i,k]+d[k,j]$ 。

模板题链接：854. Floyd求最短路 - AcWing题库

模板代码（时间复杂度n^3）

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 210, INF = 1e9;

int n, m, Q;
int d[N][N];

void floyd()
{
    for (int k = 1; k <= n; k ++ )
        for (int i = 1; i <= n; i ++ )
            for (int j = 1; j <= n; j ++ )
                d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
}

int main()
{
    scanf("%d%d%d", &n, &m, &Q);

    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= n; j ++ )
            if (i == j) d[i][j] = 0;
            else d[i][j] = INF;

    while (m -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        d[a][b] = min(d[a][b], c);
    }

    floyd();

    while (Q -- )
    {
        int a, b;
        scanf("%d%d", &a, &b);

        int t = d[a][b];
        if (t > INF / 2) puts("impossible");
        else printf("%d\n", t);
    }

    return 0;
}