Hdu1757 - A Simple Math Problem - 矩阵快速幂

#include<stdio.h>
#include<string.h>
int n,m;
long long k;
struct JZ
{
int jz[50][50];
};
JZ mul(JZ a,JZ b)
{
JZ ans;
int i,j,k;
for(i=0;i<=9;i++)
{
for(j=0;j<=9;j++)
{
ans.jz[i][j]=0;
for(k=0;k<=9;k++)
{
ans.jz[i][j]+=(a.jz[i][k]%m)*(b.jz[k][j]%m);
}
ans.jz[i][j]=ans.jz[i][j]%m;
}
}
return ans;
}
JZ pow(JZ c,long long d)
{
JZ ans;
int i,j;
for(i=0;i<=9;i++)
{
for(j=0;j<=9;j++)
{
if(j==i)
{
ans.jz[i][j]=1;
}
else
{
ans.jz[i][j]=0;
}
}
}
while(d)
{
if(d%2==1)
{
ans=mul(ans,c);
}
c=mul(c,c);
d=d/2;
}
return ans;
}
int main()
{
int i,j;
while(scanf("%I64d%d",&k,&m)!=EOF)
{
JZ f;
for(i=0;i<=9;i++)
{
for(j=0;j<=9;j++)
{
if(i==0)
{
f.jz[i][j]=j;
}
else
{
f.jz[i][j]=0;
}
}
}
JZ simple;
int x;
for(i=9;i>=0;i--)
{
scanf("%d",&x);
for(j=0;j<=9;j++)
{
if(j==9)
{
simple.jz[i][j]=x;
}
else if(j==i-1)
{
simple.jz[i][j]=1;
}
else
{
simple.jz[i][j]=0;
}
}
}
if(k>9)
{
simple=pow(simple,k-9);
f=mul(f,simple);
printf("%d\n",f.jz[0][9]%m);
}
else
{
printf("%d\n",f.jz[0][k]%m);
}
}
}