hdu4965矩阵快速幂

这题不能直接按常规做啊，因为数组根本就开不下，转换思维A（B*A）^(n*n-1)B

这样的话数组B*A就是10*10了，然后快速幂就行了

刚开始数组都开小了，tle，还找了半天bug。。。还有就是定义三个struct真累。。。

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<stack>
#include<vector>
#include<cstdio>
#include<iomanip>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define pi acos(-1)
#define ll long long
#define mod 1000000007
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#pragma comment(linker, "/STACK:1024000000,1024000000")

using namespace std;

const double g=10.0,eps=1e-9;
const int N=1000+5,maxn=1<<10+5,inf=0x3f3f3f3f;

struct NodeA{
    ll row,col;
    ll a[N][10];
};
struct NodeB{
    ll row,col;
    ll a[10][N];
};
struct Node{
    ll row,col;
    ll a[10][10];
};
Node mul(Node x,Node y)
{
    Node ans;
    ans.row=x.row,ans.col=y.col;
    memset(ans.a,0,sizeof ans.a);
    for(ll i=0;i<x.row;i++)
        for(ll j=0;j<x.col;j++)
            for(ll k=0;k<y.col;k++)
                ans.a[i][k]=(ans.a[i][k]+x.a[i][j]*y.a[j][k])%6;
    return ans;
}
Node quick_mul(Node x,ll n)
{
    Node ans;
    ans.row=x.row,ans.col=x.col;
    memset(ans.a,0,sizeof ans.a);
    for(ll i=0;i<ans.col;i++)ans.a[i][i]=1;
    while(n){
        if(n&1)ans=mul(ans,x);
        x=mul(x,x);
        n>>=1;
    }
    return ans;
}
int main()
{

    ios::sync_with_stdio(false);
    cin.tie(0);
 //   cout<<setiosflags(ios::fixed)<<setprecision(2);
    ll n,m;
    while(cin>>n>>m,n&&m){
        NodeA A;
        A.row=n,A.col=m;
        for(ll i=0;i<A.row;i++)
            for(ll j=0;j<A.col;j++)
                cin>>A.a[i][j];
        NodeB B;
        B.row=m,B.col=n;
        for(ll i=0;i<B.row;i++)
            for(ll j=0;j<B.col;j++)
                cin>>B.a[i][j];
        Node C;
        C.row=m,C.col=m;
        memset(C.a,0,sizeof C.a);
        for(ll i=0;i<B.row;i++)
            for(ll j=0;j<B.col;j++)
                for(ll k=0;k<A.col;k++)
                    C.a[i][k]=(C.a[i][k]+B.a[i][j]*A.a[j][k])%6;
        C=quick_mul(C,n*n-1);
        NodeA P;
        P.row=n,P.col=m;
        memset(P.a,0,sizeof P.a);
        for(ll i=0;i<A.row;i++)
            for(ll j=0;j<A.col;j++)
                for(ll k=0;k<C.col;k++)
                    P.a[i][k]=(P.a[i][k]+A.a[i][j]*C.a[j][k])%6;
        ll ans=0,s[N];
        for(ll i=0;i<P.row;i++)
        {
            memset(s,0,sizeof s);
            for(ll j=0;j<P.col;j++)
                for(ll k=0;k<B.col;k++)
                {
                    s[k]=(s[k]+P.a[i][j]*B.a[j][k])%6;
                    if(j==P.col-1)ans+=s[k];
                }
        }
        cout<<ans<<endl;
    }
    return 0;
}