m*m的矩阵的n次快速幂

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e4;
int n,m;
int mod =1e4;
int temp[maxn][maxn];
int res[maxn][maxn],a[maxn][maxn];
void mul(int a[][maxn],int b[][maxn]){
 memset(temp,0,sizeof(temp));
 for(int i = 0 ; i < m ; i++)
  for(int j = 0 ; j < m ; j++)
   for(int k = 0 ; k < m ; k++)
    temp[i][j]= (temp[i][j]+a[i][k]*b[k][j]%mod)%mod;
 for(int i = 0 ; i < m ; i++)
  for(int j = 0 ; j < m ; j++)
   a[i][j]= temp[i][j];
 return;
}
void fun(int nn){
 memset(res,0,sizeof(res));
 for(int i = 0 ; i < m ; i++)
  res[i][i] = 1;
 while(nn){
  if(nn&1)
   mul(res,a);
  mul(a,a);
  nn>>=1;
 }
 return ;
}
int main(){
 cin>>n>>m;
 for(int i = 0 ; i < m ; i++)
  for(int j = 0 ; j < m ; j++)
   cin>>a[i][j];
  fun(n);
  for(int i = 0 ; i < m ; i++){
   for(int j = 0 ; j < m ; j++)
    cout<<res[i][j]<<" ";
   cout<<endl;
  }
 return 0;
}