HDU 2807 The Shortest Path(3维矩阵转2维)

本题要使满足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;
}