题意
给出一张图,每条边每天都有一个权值,如果是 0 0 0就代表没有连边。
求出从 1 1 1号节点走 M M M天到达 N N N号节点的最短路。
思路
N N N和 M M M的范围很小,所以我们可以用 O ( N 2 M ) O(N^2M) O(N2M)的动态规划。
设 f i , j f_{i,j} fi,j为从起点走 i i i天到 j j j点的最短路,转移显然。
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
int n, m;
int d[101][101][21], f[201][101];
int main() {
scanf("%d %d", &n, &m);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++) {
if (i == j) continue;
scanf("%d", &d[i][j][0]);
for (int k = 1; k <= d[i][j][0]; k++) {
scanf("%d", &d[i][j][k]);
if (!d[i][j][k]) d[i][j][k] = 707406378;
}
}
memset(f, 127 / 3, sizeof(f));
f[0][1] = 0;
for (int i = 1; i <= m; i++)
for (int j = 1; j <= n; j++)
for (int k = 1; k <= n; k++)
if (j != k && d[k][j][0])
f[i][j] = std::min(f[i][j], f[i - 1][k] + d[k][j][(i - 1) % d[k][j][0] + 1]);
if (f[m][n] == 707406378) printf("0");
else printf("%d", f[m][n]);
}