HDU 4965 Fast Matrix Calculation (矩阵快速幂取模----矩阵相乘满足结合律)

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

利用相乘的可结合性先算B*A，得到6*6的矩阵，利用矩阵快速幂取模即可水过。

#include<iostream>
#include<stdio.h>
#include<iostream>
#include<stdio.h>
#define N 1010
#define M 1010
#define K 6
using namespace std;
struct matrix
{
    int m[N][M];
    int row,col;
};
matrix A,B,C,ANS;
matrix t;
matrix &matrix_mul(matrix &a,matrix &b,int p)//a，b矩阵相乘每个数模p
{
    for(int i=0;i<a.row;i++)
            for(int j=0;j<b.col;j++)
                t.m[i][j]=0;
    t.row=a.row;
    t.col=b.col;
    for(int i=0;i<a.row;i++)
        for(int k=0;k<a.col;k++)
        {
            for(int j=0;j<b.col;j++)
                t.m[i][j]=(t.m[i][j]+a.m[i][k]*b.m[k][j])%p;
        }
    return t;
}

    matrix ans;
matrix &fast_matrixt_mul_mod(matrix &a,int b,int p)//矩阵a的b次方模p
{
    for(int i=0;i<N;i++)
        for(int j=0;j<M;j++)
        if(i==j)ans.m[i][j]=1;
        else ans.m[i][j]=0;
    ans.row=a.row;
    ans.col=a.col;
        //cout<<"=="<<endl;
    while(b)
    {
        if(b&1)
        {
            b--;
            ans=matrix_mul(ans,a,p);//ans=ans*a%m;
        }
        else
        {
            b/=2;
            a=matrix_mul(a,a,p);//a=a*a%m;
        }
    }
    return ans;
}
int main()
{
    int n,k;
    while(scanf("%d%d",&n,&k),n!=0||k!=0)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<k;j++)
                scanf("%d",&A.m[i][j]);
        }
        A.row=n;
        A.col=k;
        for(int i=0;i<k;i++)
        {
            for(int j=0;j<n;j++)
                scanf("%d",&B.m[i][j]);
        }
        B.row=k;
        B.col=n;
        C=matrix_mul(B,A,6);
        C=fast_matrixt_mul_mod(C,n*n-1,6);
        ANS=matrix_mul(A,C,6);
        ANS=matrix_mul(ANS,B,6);
        int sum=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            sum+=(ANS.m[i][j]%6);
        }
        printf("%d
",sum);
    }
    return 0;
}