解决多源多汇的最短路,时间复杂度O(n^3)
传递闭包
Flyd求最短路是基于动态规划实现的。
集合 dist[k, i, j] 表示只经过 1 ~ k 的节点,从 i 到 j 的最短路长度。
可以划分为两个子问题:经过 k 这个点和不经过 k 这个点
转移方程:dist[k, i, j] = min(dist[k - 1, i, j], dist[k - 1, i, k] + dist[k - 1, k, j]。
和背包问题相似,可以省略第一维:
dist[i, j] = min( dist[i, j], dist[i, k] + dist[k, j] )
dist 数组就是邻接矩阵。
void floyd()
{
cin >> n >> m >> k;
memset(dist, 0x3f, sizeof dist);
for(int i = 1; i <= m; i++)
{
int a, b, c; cin >> a >> b >> c;
if(a == b) continue; // 去除自环
dist[a][b] = min(dist[a][b], c); // 重边取最小边
}
for(int i = 1; i <= n; i++)
dist[i][i] = 0;
for(int k = 1; k <= n; k++)
for(int i = 1; i <= n; i++)
for(int j = 1; j <= n; j++)
dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j]);
}