HDU - 6331 Walking Plan 矩阵乘法 最短路

题意：

给你n个点m条边的有向图。有Q次询问，每次问你从s到t至少经过k条边的最短距离。

n <= 50, m<= 10000, q<= 100000, k<= 10000.

题解：

设f[i][j][k]为从i到j恰好经过了k条边的最小距离。

f[k][i][j] = min( f[x][i][t] + f[k - x][t][j] )。

注意到这个转移类似矩阵乘法。

我们用矩阵乘法f[x] * f[y] 即可代表恰好走x + y条路的任意两点间最短路径。

设g为原图的矩阵。 

则 f[i] = f[i - 1] * g。

f[10000][50][50] 为我们需要的大小。 但是每次矩阵乘复杂度n^3，整体复杂度k*n^3 = 1.25e9。

但是注意到我们可以很快的求出 恰好经过100条边， 100 * 2条边， 100 * 3条.....的矩阵，只要乘以g ^ 100次方即可。

我们设a[i] 为恰好经过i条边时的图， b[i]为恰好经过100*i条边的图， 再设c[i]为至少经过i条边的图。

这个c[i] 可以通过a[i]转移过来，方法是将原图g 作floyd算法得到g'，然后c[i] = a[i] * g'。

此时对于每次询问，我们设A = K  % 100, B = k / 100。枚举中间点i，答案即是 min(c[A][s][i] + b[B][i][t])。

代码：

#include<bits/stdc++.h>
#define pb push_back
#define pii pair<int,int>
#define mp make_pair
#define lson id<<1,l,mid
#define rson id<<1|1,mid+1,r
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define debug(x) cout<<#x<<" = "<<x<<endl;
typedef long long ll;
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXN = 1e2 + 5;
const int MOD = 1e9 + 7;

int g[55][55];

int a[MAXN][55][55];
int b[MAXN][55][55];
int d[55][55];
int f[55][55];
int n, m;

void mul (int a[][55], int b[][55], int c[][55]) {
    for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) f[i][j] = INF;
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            for (int k = 1; k <= n; k++)
                f[i][j] = min (a[i][k] + b[k][j], f[i][j]);
    for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) c[i][j] = f[i][j];
}

int main() {
#ifdef LOCAL
    freopen ("input.txt", "r", stdin);
#endif
    int T;
    cin >> T;
    while (T--) {
        scanf ("%d %d", &n, &m);
        rep(i, 1, n) rep(j, 1, n) g[i][j] = INF;
        for (int i = 1; i <= m; i++) {
            int x, y, w;
            scanf ("%d %d %d", &x, &y, &w);
            g[x][y] = min (g[x][y], w);
        }
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                a[0][i][j] = i == j ? 0 : INF;
                b[0][i][j] = i == j ? 0 : INF;
            }
        }
        rep(i, 1, 102) mul (a[i - 1], g, a[i]);
        rep(i, 1, 102) mul (b[i - 1], a[100], b[i]);
        for (int i = 1; i <= n; i++) g[i][i] = 0;
        for (int k = 1; k <= n; k++)
            for (int i = 1; i <= n; i++)
                for (int j = 1; j <= n; j++)
                    g[i][j] = min(g[i][j], g[i][k] + g[k][j]);
        for(int i = 0; i <= 102; i++) mul(b[i], g, b[i]);
        int q;
        scanf("%d", &q);
        while(q--) {
            int s, t, k;
            scanf("%d %d %d", &s, &t, &k);
            int ans = INF;
            int A = k / 100, B = k % 100;
            for(int i = 1; i <= n; i++) ans = min(ans, b[A][s][i] + a[B][i][t]);
            if(ans != INF) printf("%d\n", ans);
            else puts("-1");
        }
    }
    return 0;
}