Fast Matrix Calculation HDU - 4965

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

(A*B)^(n*n)=A * ( (B*A)^(n*n-1) ) * B

#include <bits/stdc++.h>
using namespace std;
const int mod=6;

int a[1010][10],b[10][1010],c[10][10],ans[10][10];
int x[1010][10],y[1010][1010];
int n,m;

void getmul(int u[][10],int v[][10])
{
    int t[10][10];
    int i,j,k;
    for(i=1;i<=m;i++)
    {
        for(j=1;j<=m;j++)
        {
            t[i][j]=0;
            for(k=1;k<=m;k++) t[i][j]=(t[i][j]+(u[i][k]*v[k][j])%mod)%mod;
        }
    }
    memcpy(u,t,sizeof(t));
}

void quickpow(int p)
{
    int i,j;
    memset(ans,0,sizeof(ans));
    for(i=1;i<=m;i++) ans[i][i]=1;
    while(p>0)
    {
        if(p%2) getmul(ans,c);
        getmul(c,c),p/=2;
    }
}

int main()
{
    int i,j,k,sum;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==0&&m==0) break;
        for(i=1;i<=n;i++) for(j=1;j<=m;j++) scanf("%d",&a[i][j]);
        for(i=1;i<=m;i++) for(j=1;j<=n;j++) scanf("%d",&b[i][j]);
        for(i=1;i<=m;i++)//mn*nm
        {
            for(j=1;j<=m;j++)
            {
                c[i][j]=0;
                for(k=1;k<=n;k++) c[i][j]=(c[i][j]+(b[i][k]*a[k][j])%mod)%mod;
            }
        }
        quickpow(n*n-1);
        for(i=1;i<=n;i++)//nm*mm
        {
            for(j=1;j<=m;j++)
            {
                x[i][j]=0;
                for(k=1;k<=m;k++) x[i][j]=(x[i][j]+(a[i][k]*ans[k][j])%mod)%mod;
            }
        }
        for(i=1;i<=n;i++)//nm*mn
        {
            for(j=1;j<=n;j++)
            {
                y[i][j]=0;
                for(k=1;k<=m;k++) y[i][j]=(y[i][j]+(x[i][k]*b[k][j])%mod)%mod;
            }
        }
        sum=0;
        for(i=1;i<=n;i++) for(j=1;j<=n;j++) sum+=y[i][j];
        printf("%d\n",sum);
    }
    return 0;
}