HDU 4965 Fast Matrix Calculation

矩阵快速幂 肯定是这个没问题了。

一上来我就贴了模板。 可是一看矩阵最大是 1000*1000的。 结构体内 数组开不开这么大啊。 明显模板不合适了。。

然后看了看上面的条件。 发现（AB）^(N*N)  如果N 为2 的话  就是ABABABAB  结合律 A（BA）（BA）（BA）B

这样就是 BA 的快速幂 结果左乘 A 右乘 B。 并且 BA 是 10*10 的， 就可以用矩阵快速幂了。

还有矩阵求和的公式。  第一层循环是 前面矩阵的行数 第二层循环是 后面矩阵的 行数  第三层循环是前面矩阵的列数。

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <string>
#include <map>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <cctype>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define MAXN 1000+10
#define INF 1<<30
#define mod 6
int N,M;
struct Matrix
{
    int n,m;
    int a[20][20];
    void clear(int x=0,int y=0)
    {
        n=x;
        m=y;
        memset(a,0,sizeof(a));
    }
};
Matrix cc(Matrix a,Matrix b)
{
       Matrix tmp;
       tmp.clear(M,M);
       for(int i = 0; i < M; ++i)
            for(int j = 0; j < M; ++j)
                for(int k = 0; k < M; ++k)
                {
                    tmp.a[i][j]+=a.a[i][k]*b.a[k][j];
                    tmp.a[i][j]%=mod;
                }
       return tmp;
}

Matrix _Pow(Matrix mat,int n)
{
    Matrix res;
    res.n = 10,res.m = 10;
    for(int i = 0; i < 10 ;i++)
        for(int j = 0; j < 10; j++){
            if(i == j)
            res.a[i][i] = 1;
            else
            res.a[i][j] = 0;

        }
    while(n)
    {
        if(n&1) res=cc(res,mat);
        mat=cc(mat,mat);
        n=n>>1;
    }
    return res;
}
Matrix BA;
int A[MAXN][MAXN],B[MAXN][MAXN],C[MAXN][MAXN],L[MAXN][MAXN],LL[MAXN][MAXN];
int main (){
    while(scanf("%d%d",&N,&M) != EOF){
        memset(LL,0,sizeof(LL));
        memset(L,0,sizeof(L));
        memset(A,0,sizeof(A));
        memset(B,0,sizeof(B));
        memset(C,0,sizeof(C));
        //  1 是 前的行 2 是后的行 3是 前的列
        if(N == 0 && M == 0)
            break;
        for(int i = 0; i < N; i++){
            for(int j = 0; j < M; j++)
                scanf("%d",&A[i][j]);
        }
        for(int i = 0; i < M; i++){
            for(int j = 0; j < N; j++){
                scanf("%d",&B[i][j]);
            }
        }

        for(int i=0; i< M; ++i){
            for(int j=0; j < M; ++j)
                for(int k=0; k < N; ++k)
                {
                    C[i][j] += B[i][k]*A[k][j];
                    C[i][j] %= mod;
                    BA.a[i][j] = C[i][j];
                }
        }
        Matrix M_;
        M_ = _Pow(BA,N*N-1);
        for(int i = 0; i< N; ++i){
            for(int j = 0; j < M; ++j)
                for(int k = 0; k < M; ++k)
                {
                    L[i][j] += A[i][k]*M_.a[k][j];
                    L[i][j] %= mod;
                }
        }
        for(int i = 0; i< N; ++i){
            for(int j = 0; j < N; ++j)
                for(int k = 0; k < M; ++k)
                {
                    LL[i][j] += L[i][k]*B[k][j];
                    LL[i][j] %= mod;
                }
        }
        int sum = 0;
        for(int i = 0; i < N; i++){
            for(int j = 0; j < N; j++){
                sum += LL[i][j];
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}