HDU 6331 Walking Plan（分块 动态规划）

题目链接：Walking Plan

题意

给一个 $n$ 个节点 $m$ 条边的有向图，第 $i$ 条边的两个端点为 $u_i,v_i$，边的长度为 $w_i$，$q$ 次询问，每次询问从节点 $s$ 到 $t$ 至少走过 $k$ 条路径的最小距离。

输入

第一行包含一个整数 $T~(1\leq T\leq10)$，接下去有 $T$ 组数据，每组数据第一行为两个整数 $n,m~(2\leq n\leq50,1\leq m\leq10^4)$，接下去 $m$ 行每行三个整数 $u_i,v_i,w_i~(1\leq u_i,v_i\leq n,u_i\neq v_i,1\leq w_i\leq10^4)$，接着为一个整数 $q~(1\leq q\leq10^5)$，接下去 $q$ 行每行三个整数 $s_i,t_i,k_i~(1\leq s_i,t_i\leq n,1\leq k\leq10^4)$。

输出

对于每次询问，输出最短路径长度，如果无法从节点 $s_i$ 到达 $t_i$，则输出 $-1$。

样例

输入
2 3 3 1 2 1 2 3 10 3 1 100 3 1 1 1 1 2 1 1 3 1 2 1 1 2 1 1 2 1 1
输出
111 1 11 -1

题解

定义 $G[i][j]$ 为从节点 $i$ 恰好经过 $1$ 步到达节点 $j$ 的最短距离（即原图按输入取最小值），无法到达设为 $\infty$，$dis[k][i][j]$ 表示从 $i$ 点出发恰好经过 $k$ 步到达 $j$ 点的最短路径，则有递推式：

$$dis[0][i][j]=\begin{cases} 0&i=j\\ \infty&i\neq j \end{cases}\\ dis[k][i][j]=\min(dis[k-1][i][u]+G[u][j]),u\in[1,n],k\neq0\\$$ 定义 $A[x][i][j]$ 表示从 $i$ 恰好经过 $100x$ 步到达 $j$ 的最短距离，$B[x][i][j]$ 表示从 $i$ 最少经过 $x$ 次到达 $j$ 的最短距离，这样 $k$ 就可以分为 $\lfloor\frac{k}{100}\rfloor$ 和 $k\%100$ 两个部分，通过枚举中转点得到答案：

$$ans=\min(A[\lfloor\frac{k}{100}\rfloor][i][u]+B[k\%100][u][j]),u\in[1,n]$$ 第一部分可以直接从上面的递推式得到，第二部分可以在原图上令 $G[i][i]=0,i\in[1,n]$ 后跑一遍 $floyd$，再将从 $i$ 恰好经过 $k$ 步到达 $j$ 的最短距离距离 $dis[k][i][j]$ 通过 $G[i][j]$ 转化为 $B[x][i][j]$：
$$B[k][i][j]=\min(dis[k][i][u],G[u][j]),u\in[1,n]$$

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <bitset>
#include <algorithm>
#include <functional>
#include <iomanip>
using namespace std;

#define LL long long
const int Size = 51;
const int maxn = 101;
int n;
struct Matrix {
    int num[Size][Size];

    void Init() {
        for(int i = 1; i <= n; ++i) {
            memset(num[i], 0x3f, sizeof(int) * (n + 1));
        }
    }

    void Set_zero() {
        for(int i = 1; i <= n; ++i) {
            num[i][i] = 0;
        }
    }

    void operator=(const Matrix &m) {
        for(int i = 1; i <= n; ++i) {
            memcpy(num[i], m.num[i], sizeof(int) * (n + 1));
        }
    }

    void Combine(const Matrix &m, Matrix &ans) {
        Matrix ret;
        ret.Init();
        for(int i = 1; i <= n; ++i) {
            for(int j = 1; j <= n; ++j) {
                for(int k = 1; k <= n; ++k) {
                    ret.num[i][j] = min(ret.num[i][j], num[i][k] + m.num[k][j]);
                }
            }
        }
        ans = ret;
    }

    void floyd() {
        for(int k = 1; k <= n; ++k) {
            for(int i = 1; i <= n; ++i) {
                for(int j = 1; j <= n; ++j) {
                    num[i][j] = min(num[i][j], num[i][k] + num[k][j]);
                }
            }
        }
    }
};

int T, m, u, v, dis, q, k, INF, ans;
Matrix A[maxn], B[maxn], G;

int main() {
    #ifdef LOCAL
    freopen("test.txt", "r", stdin);
//    freopen("test1.out", "w", stdout);
    #endif // LOCAL
    ios::sync_with_stdio(false);

    memset(&INF, 0x3f, sizeof(int));
    scanf("%d", &T);
    while(T--) {
        scanf("%d%d", &n, &m);
        G.Init();
        for(int i = 0; i < m; ++i) {
            scanf("%d%d%d", &u, &v, &dis);
            G.num[u][v] = min(G.num[u][v], dis);
        }
        A[0].Init();
        A[0].Set_zero();
        B[0].Init();
        B[0].Set_zero();
        for(int i = 1; i < maxn; ++i) {
            B[i - 1].Combine(G, B[i]);
        }
        for(int i = 1; i < maxn; ++i) {
            A[i - 1].Combine(B[100], A[i]);
        }
        G.Set_zero();
        G.floyd();
        for(int i = 0; i < maxn; ++i) {
            G.Combine(B[i], B[i]);
        }
        scanf("%d", &q);
        while(q--) {
            scanf("%d%d%d", &u, &v, &k);
            ans = INF;
            for(int i = 1; i <= n; ++i) {
                ans = min(ans, A[k / 100].num[u][i] + B[k % 100].num[i][v]);
            }
            if(ans == INF) {
                printf("-1\n");
            } else {
                printf("%d\n", ans);
            }
        }
    }

    return 0;
}