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-;

const int N=+,maxn=<<+,inf=0x3f3f3f3f;

struct NodeA{

    ll row,col;

    ll a[N][];

};

struct NodeB{

    ll row,col;

    ll a[][N];

};

struct Node{

    ll row,col;

    ll a[][];

};

Node mul(Node x,Node y)

{

    Node ans;

    ans.row=x.row,ans.col=y.col;

    memset(ans.a,,sizeof ans.a);

    for(ll i=;i<x.row;i++)

        for(ll j=;j<x.col;j++)

            for(ll k=;k<y.col;k++)

                ans.a[i][k]=(ans.a[i][k]+x.a[i][j]*y.a[j][k])%;

    return ans;

}

Node quick_mul(Node x,ll n)

{

    Node ans;

    ans.row=x.row,ans.col=x.col;

    memset(ans.a,,sizeof ans.a);

    for(ll i=;i<ans.col;i++)ans.a[i][i]=;

    while(n){

        if(n&)ans=mul(ans,x);

        x=mul(x,x);

        n>>=;

    }

    return ans;

}

int main()

{

    ios::sync_with_stdio(false);

    cin.tie();

 //   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=;i<A.row;i++)

            for(ll j=;j<A.col;j++)

                cin>>A.a[i][j];

        NodeB B;

        B.row=m,B.col=n;

        for(ll i=;i<B.row;i++)

            for(ll j=;j<B.col;j++)

                cin>>B.a[i][j];

        Node C;

        C.row=m,C.col=m;

        memset(C.a,,sizeof C.a);

        for(ll i=;i<B.row;i++)

            for(ll j=;j<B.col;j++)

                for(ll k=;k<A.col;k++)

                    C.a[i][k]=(C.a[i][k]+B.a[i][j]*A.a[j][k])%;

        C=quick_mul(C,n*n-);

        NodeA P;

        P.row=n,P.col=m;

        memset(P.a,,sizeof P.a);

        for(ll i=;i<A.row;i++)

            for(ll j=;j<A.col;j++)

                for(ll k=;k<C.col;k++)

                    P.a[i][k]=(P.a[i][k]+A.a[i][j]*C.a[j][k])%;

        ll ans=,s[N];

        for(ll i=;i<P.row;i++)

        {

            memset(s,,sizeof s);

            for(ll j=;j<P.col;j++)

                for(ll k=;k<B.col;k++)

                {

                    s[k]=(s[k]+P.a[i][j]*B.a[j][k])%;

                    if(j==P.col-)ans+=s[k];

                }

        }

        cout<<ans<<endl;

    }

    return ;

}