HDU 6331 Problem M. Walking Plan（分块）

Description

给出一个$n$个点$m$条边的无向图，$q$次查询，每次查询从$u$点出发至少走$k$步到达$v$点的最短路径长度

Input

第一行一整数$T$表示用例组数，每组用例首先输入两个整数$n,m$表示点数和边数，之后$m$行每行三个整数$u,v,w$表示$u,v$之间有一条权值为$w$的无向边，之后输入一整数$q$表示查询次数，最后$q$行每行三个整数$u,v,k$表示查询从$u$点出发至少走$k$步到达$v$点的最短路径长度

$(1\le T\le 10,2\le n\le 50,1\le m\le 10^4,1\le w_i\le 10^4,1\le q\le 10^5,1\le k\le 10^4)$

Output

对于每组查询，输出结果，如果两点不可达则输出$-1$

Sample Input

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

Sample Output

111
1
11
-1

Solution

假设该有向图的邻接矩阵为$A$，那么查询$u\ v\ k$的答案即为$min(A^k(u,v),...,A^{k+n}(u,v))$，由于两点间路径至多经过$n$个点，故超过$n$步再多走答案不会更优，但是无论是预处理$A^k$还是直接矩阵快速幂求复杂度都不够，考虑令$k=100\cdot x+y,x\ge 0,y>0$，那么我们只要预处理出$A^y,B^x=A^{100x}$，每次查询即可$O(n)$的求出这两个矩阵的第$u$行乘第$v$列的结果，此时单组查询的复杂度是$O(n^2)$，进一步的，对$A^y$求后缀最小值即可$O(n)$完成单组查询

Code

#include<cstdio>
#include<algorithm>
using namespace std;
typedef int M[51][51];
#define INF 1000000000
M A[155],B[155];
int T,n,m,q;
void Mul(M A,M B,M &C)
{
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            C[i][j]=INF;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(A[i][j]!=INF)
                for(int k=0;k<n;k++)
                    if(B[j][k]!=INF)
                        C[i][k]=min(C[i][k],A[i][j]+B[j][k]);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                A[1][i][j]=INF;
        while(m--)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            u--,v--;
            A[1][u][v]=min(A[1][u][v],w); 
        }
        for(int i=2;i<=100+n;i++)Mul(A[i-1],A[1],A[i]);
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                B[0][i][j]=(i==j?0:INF);
        for(int i=1;i<=100;i++)Mul(B[i-1],A[100],B[i]);
        for(int i=100+n-1;i>=1;i--)
        {
            for(int j=0;j<n;j++)
                for(int k=0;k<n;k++)
                    A[i][j][k]=min(A[i][j][k],A[i+1][j][k]);
        }
        scanf("%d",&q);
        while(q--)
        {
            int u,v,k;
            scanf("%d%d%d",&u,&v,&k);
            u--,v--;
            int x=(k-1)/100,y=k-x*100;
            int ans=INF;
            for(int i=0;i<n;i++)
                ans=min(ans,B[x][u][i]+A[y][i][v]);
            if(ans!=INF)printf("%d\n",ans);
            else printf("-1\n");
        }
    }
    return 0;
}