本题要使满足A*B=C的A,C间有一条边,关键是怎么高效计算矩阵乘法。
可在等式两边同时乘上一个长为m,宽为1,且数值全为1的矩阵,将两侧均转换为 A‘ * B = C'
只需要预处理一遍即可将矩阵乘法的时间复杂度从O(n^3)降至O(n^2)
#include <cstdio>
using namespace std;
const int maxn =80+5;
const int inf = 100000000;
int a[maxn][maxn][maxn],b[maxn][maxn],d[maxn][maxn];
int n,m;
void compare(int x,int y)
{
for(int i=1;i<=m;i++)
if(b[0][i]!=b[y][i]) return ;
d[x][y]=1;
}
void create(int x,int y)
{
for(int i=1;i<=m;i++){
b[0][i]=0;
for(int j=1;j<=m;j++)
b[0][i]+=b[y][j]*a[x][i][j];
}
for(int i=1;i<=n;i++)
if(i!=x&&i!=y) compare(x,i);
}
void floyd(){
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(d[i][j]>d[i][k]+d[k][j])
d[i][j]=d[i][k]+d[k][j];
}
int main()
{
while(~scanf("%d%d",&n,&m),n||m){
for(int i=1;i<=n;i++){
d[i][i]=0;
for(int j=1;j<i;j++)
d[i][j]=d[j][i]=inf;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
b[i][j]=0;
for(int k=1;k<=m;k++){
scanf("%d",&a[i][j][k]);
b[i][j]+=a[i][j][k];
}
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(i!=j) create(i,j);
floyd();
int k;
scanf("%d",&k);
while(k--){
int x,y;
scanf("%d%d",&x,&y);
if(d[x][y]!=inf) printf("%d\n",d[x][y]);
else printf("Sorry\n");
}
}
return 0;
}