基于动态规划的原理,f[k][i][j]
表示通过前k
个点从i
到j
的距离,转移方程f[k][i][j] = f[k-1][i][k] + f[k-1][k][j]
,含义为用通过前k-1
个点从i
到达k
的距离与通过前k-1
个点从k
到达j
的距离
之和更新从前k
个点从i
到j
的距离。
第一维度可以忽略。
一般采用邻接矩阵来保存边,i==j
的情况为
0
0
0,其余为无穷。
注意存在负边权时,当d>INF/2
,说明不可达。
#include<bits/stdc++.h>
using namespace std;
const int N = 100;
int n, e[N][N];
int main()
{
cin >> n;
for (int i = 1;i <= n;i ++)
for (int j = 1;j <= n;j ++)
cin >> e[i][j],
e[i][j] = (e[i][j] == 0 && i != j? 0x3f3f3f3f : e[i][j]);
for (int k = 1;k <= n;k ++)
for (int i = 1;i <= n;i ++)
for (int j = 1;j <= n;j ++)
e[i][j] = min (e[i][j], e[i][k] + e[k][j]);
for (int i = 1;i <= n;i ++, cout << endl)
for (int j = 1;j <= n;j ++)
cout << (e[i][j] == 0x3f3f3f3f ? -1 : e[i][j]) << ' ';
return 0;
}