Description
给出一个 n n 个点条边的无向图, q q 次查询,每次查询从点出发至少走 k k 步到达点的最短路径长度
Input
第一行一整数 T T 表示用例组数,每组用例首先输入两个整数表示点数和边数,之后 m m 行每行三个整数表示 u,v u , v 之间有一条权值为 w w 的无向边,之后输入一整数表示查询次数,最后 q q 行每行三个整数表示查询从 u u 点出发至少走步到达 v v 点的最短路径长度
Output
对于每组查询,输出结果,如果两点不可达则输出 −1 − 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 A ,那么查询的答案即为 min(Ak(u,v),...,Ak+n(u,v)) m i n ( A k ( u , v ) , . . . , A k + n ( u , v ) ) ,由于两点间路径至多经过 n n 个点,故超过步再多走答案不会更优,但是无论是预处理 Ak A k 还是直接矩阵快速幂求复杂度都不够,考虑令 k=100⋅x+y,x≥0,y>0 k = 100 ⋅ x + y , x ≥ 0 , y > 0 ,那么我们只要预处理出 Ay,Bx=A100x A y , B x = A 100 x ,每次查询即可 O(n) O ( n ) 的求出这两个矩阵的第 u u 行乘第列的结果,此时单组查询的复杂度是 O(n2) O ( n 2 ) ,进一步的,对 Ay A y 求后缀最小值即可 O(n) 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;
}