这题不能直接按常规做啊，因为数组根本就开不下，转换思维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; }