首页 > 代码库 > Floyd 算法求多源最短路径

Floyd 算法求多源最短路径

Floyd算法:

  Floyd算法用来找出每对顶点之间的最短距离,它对图的要求是,既可以是无向图也可以是有向图,边权可以为负,但是不能存在负环.

基本算法:

  Floyd算法基于动态规划的思想,以 u 到 v 的最短路径至少经过前 k 个点为转移状态进行计算,通过 k 的增加达到寻找最短路径的目的.当 k 增加 1 时,最短路径要么不边,如果改变,必经过第 k 各点,也就是说当起点 u 到第 k 个点的最短距离加上第 k 个点到终点 v 的最短路径小于不经过第 k 个节点的最优最短路经长度的时候更新 u 到 v 的最短距离. 当 k = n 时, u 到 v 的最短路径就确定了. 

伪代码:

  图的存储用邻接矩阵 gra[][] 来记录,如果 u 与 v 之间没有边直接相连,则 gra[u][v] = INF; dist[][] 记录最终的最短路. pre[i][j] 存储 i 到 j 路径中 i 的后一个节点.

  1): 初始化:将 gra 中的数据复制到 dist 中作为每对顶点间的最短路的除值, pre[i][j] = j;

  2): k 从 1 到 n 循环 n 次, 每次循环中枚举图中不同的两点 u, v, 如果 dist[u][v] > dist[u][k] + dist[k][v], 则更新 dist[u][v] = dist[u][k] + dist[k][v], 更新 pre[u][v] = pre[u][k].

  3): 最后 dist[u][v] 数组中存储的就是 u 到 v 的最短距离, u 到 v 的路径, 则可以按照顺序查找就好了.

以图为例:

有一个如下的无向图, “D” 表表示最短路表, “P” 表表示最短路径表:

技术分享

 

假设现在每对顶点之间的路径只允许经过点 “1” , 则更新后的每对顶点之间的距离:

技术分享

 

这里看到点 “2” 到点 “3” 的距离经过点 “1” 得到了更新,同时更新了用于记录路径的 P 数组.

 

第二步,允许每对顶点之间的最短路径经过点 “1” 和点 “2”,则更新后的数组为:

技术分享

 

可以看到得到更新的路径为:

1 ---> 4, 经过点 “2” 得到更新

1 ---> 5, 经过点 “2” 得到更新

3 ---> 5. 经过点 “1 --- > 2” 得到更新

 

第三步: 允许经过点 “1”, “2” 和点 “3” 则更新后的数组为:

 技术分享

这则说明,上一步的最短路径不需要更新.

 

第四步, 允许经过点 “1”, “2” , “3” 和点 “4” 则更新后的数组为:

技术分享

 

可以看到 3 ---> 5 的路径经过点 “4” 得到了更新(原先是 3 ---> 1 ---> 2 ---> 5, w = 9)

 

第五步, 允许任意两点之间的最短路径可以经过全部点,则更新后的数组为:

技术分享

这次得到更新的路径为:

1 ---> 4 的路径. 更新为 “1 ---> 2 ---> 5 ---> 4, w = 5” (原路径为 1 ---> 2 ---> 4, w = 7)

2 ---> 3 的路径. 更新为 “2 ---> 5 ---> 4 ---> 3, w = 7” (原路经为 2 ---> 1 ---> 3, w = 8)

2 ---> 4 的路径. 更新为 “2 --> 5 --> 4, w = 2” (原路径为 2 ---> 4, w = 4)

无向图反之亦然.

 

至此最短路径就寻找完毕. dist[i][j] 数组里面保存的就是 i 到 j 的最短距离.如果要查寻路径, 则按照查数组 pre 就好.比如查询 “2” 到 “3” 的路径:

则寻找     pre[2][3] = 5,  2 ---> 5

继续寻找  pre[5][3] = 4,  2 ---> 5 ---> 4

继续寻找  pre[4][3] = 3, 2 ---> 5 ---> 4 ---> 3

由于此时 i = j = 3, 则 “2” 到 “3” 的最短路径已找到为: 2 ---> 5 ---> 4 ---> 3

 

代码( 时间复杂度O(N3) ):

 1 #include <bits/stdc++.h> 2  3 typedef long long LL; 4 const int MAXN = 100; 5 const int INF = 0x3f3f3f3f; 6 using namespace std; 7  8 int pre[MAXN + 3][MAXN + 3], dist[MAXN + 3][MAXN + 3]; //pre 储存路径 dist最短距离 9 void floyd(int n, int gra[][MAXN + 3]) {10     for(int i = 1; i <= n; i++) for(int j = 1; j <= n; j++) dist[i][j] = gra[i][j], pre[i][j] = j;  //初始化11     for(int k = 1; k <= n; k++) {   //尝试经过 k 个点对每对顶点之间的距离进行更新12         for(int i = 1; i <= n; i++) {13             for(int j = 1; j <= n; j++) {14                 if(dist[i][k] != INF && dist[k][j] != INF && dist[i][k] + dist[k][j] < dist[i][j]) {15                     dist[i][j] = dist[i][k] + dist[k][j];16                     pre[i][j] = pre[i][k];17                 }18             }19         }20     }21 }22 23 int pfpath(int u, int v) { //打印最短路径24     while(u != v) {25         cout << u  << " ";26         u = pre[u][v];27     }28     cout << u << endl;29 }30 31 int gra[MAXN + 3][MAXN + 3];32 int main() {33     int n, m;34     while(cin >> n >> m){ // n 个点,  m 条边35         for(int i = 0; i <= n; i++) for(int j = -1; j <= n; j++){36             gra[i][j] = (i == j ? 0 : INF);37         }38         for(int i = 0; i < m; i++) {39             int u, v, w; cin >> u >> v >> w;40             gra[u][v] = gra[v][u] = w;  //无向图41         }42         floyd(n, gra);43     }44     return 0;45 }

 参考资料《图论及应用》

Floyd 算法求多源最短路径