hdu 2807 The Shortest Path

http://acm.hdu.edu.cn/showproblem.php?pid=2807

 第一次做矩阵乘法，没有优化超时，看了别人的优化的矩阵乘法，就过了。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define maxn 100
using namespace std;
const int inf=1<<28;

int n,m,t1,x,y;
int g[maxn][maxn];
int d[maxn][maxn];
int aa[maxn][maxn][maxn];

void deal(int a,int b)
{
    for(int i=1; i<=m; i++)
    {
        d[0][i]=0;
        for(int j=1; j<=m; j++)
        {
            d[0][i]+=(aa[a][i][j]*d[b][j]);
        }
    }
    for(int i=1; i<=n; i++)
    {
        if(i==a||i==b) continue;
        bool flag=false;
        for(int k=1; k<=m; k++)
        {
            if(d[0][k]!=d[i][k])
            {
                flag=true;
                break;
            }
        }
        if(!flag) g[a][i]=1;
    }
}


int main()
{
    while(cin>>n>>m)
    {
        if(n==0&&m==0) break;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m; j++)
            {
                d[i][j]=0;
                for(int k=1; k<=m; k++)
                {
                    scanf("%d",&aa[i][j][k]);
                    d[i][j]+=(aa[i][j][k]*k);
                }
            }
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=i+1; j<=n; j++)
            {
                g[i][j]=g[j][i]=inf;
            }
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                if(i==j) continue;
                deal(i,j);
            }
        }
        for(int k=1; k<=n; k++)
        {
            for(int i=1; i<=n; i++)
            {
                for(int j=1; j<=n; j++)
                {
                    if(g[i][j]>g[i][k]+g[k][j])
                    {
                        g[i][j]=g[i][k]+g[k][j];
                    }
                }
            }
        }
        scanf("%d",&t1);
        for(int i=0; i<t1; i++)
        {
            scanf("%d%d",&x,&y);
            if(g[x][y]!=inf) printf("%d
",g[x][y]);
            else printf("Sorry
");
        }
    }
    return 0;
}