HDU 1757 矩阵快速幂

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cmath>

using namespace std;
int m;
struct matrix
{
long long mat[20][20];
matrix()
{
memset(mat,0,sizeof(mat));
}
};
matrix mul(matrix A,matrix B)
{
matrix C;
for(int i=1;i<=10;i++)
{
for(int j=1;j<=10;j++)
{
for(int k=1;k<=10;k++)
{
C.mat[i][j]=(C.mat[i][j]+A.mat[i][k]*B.mat[k][j]%m)%m;
}
}
}
return C;
}
matrix quickmi(matrix A,long long n)
{
matrix B;
for(int i=1;i<=10;i++)
B.mat[i][i]=1;
while(n>0)
{
if(n&1) B=mul(A,B);
A=mul(A,A);
n=n/2;
}
return B;

}
int main()
{

long long k;
while(scanf("%I64d %d",&k,&m)!=EOF)
{
if(k>9){
k=k-9;
matrix A;
for(int i=1;i<=10;i++)
{
scanf("%I64d",&A.mat[1][i]);
A.mat[i+1][i]=1;
}
A=quickmi(A,k);
printf("%I64d\n",(9*A.mat[1][1]%m+8*A.mat[1][2]%m+7*A.mat[1][3]%m+6*A.mat[1][4]%m+5*A.mat[1][5]%m+4*A.mat[1][6]%m+3*A.mat[1][7]%m+2*A.mat[1][8]%m+1*A.mat[1][9]%m)%m);
}
else
{
printf("%I64d\n",k%m);
}
}
return  0;
}