hdu 1757 A Simple Math Problem
赤果果的矩阵
#include<stdio.h>
#include<string.h>
struct matrix
{
int m[10][10];
};
int mod;
matrix multi(const matrix &a,const matrix &b)
{
matrix ans;
memset(&ans,0,sizeof(ans));
for(int i=0;i<10;i++)
for(int k=0;k<10;k++)
if(a.m[i][k])
{
for(int j=0;j<10;j++)
{
ans.m[i][j]+=a.m[i][k]*b.m[k][j];
ans.m[i][j]%=mod;
}
}
return ans;
}
matrix pow(matrix a,int k)
{
matrix ans;
memset(&ans,0,sizeof(ans));
for(int i=0;i<10;i++) ans.m[i][i]=1;
while(k)
{
if(k&1) ans=multi(ans,a);
a=multi(a,a);
k>>=1;
}
return ans;
}
int main()
{
matrix a;
int n;
while(scanf("%d%d",&n,&mod)!=EOF)
{
memset(&a,0,sizeof(a));
for(int i=0;i<10;i++)
scanf("%d",&a.m[0][i]);
if(n<10)
printf("%d\n",n);
else
{
for(int i=1;i<10;i++)
a.m[i][i-1]=1;
a=pow(a,n-9);
int ans=0;
for(int i=0;i<10;i++)
{
ans+=a.m[0][i]*(10-i-1);
ans%=mod;
}
printf("%d\n",ans);
}
}
return 0;
}