[HDU] A Simple Math Problem
矩阵快速幂裸题,可以当做模板题练手。
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cmath>
using namespace std;
const int N = 15;
int m;
struct matrix{
int M[N][N];
int l,r;
void operator * (matrix x){
int i,j,k; matrix t;
for(i=1;i<=l;i++)
for(j=1;j<=r;j++) t.M[i][j]=0;
for(i=1;i<=r;i++)
for(j=1;j<=x.l;j++)
for(k=1;k<=l;k++)
t.M[j][i]=(t.M[j][i]+M[k][i]*x.M[j][k]) % m;
for(i=1;i<=l;i++)
for(j=1;j<=r;j++) M[i][j]=t.M[i][j];
}
}c,res;
int main(){
int i,j,n,k;
while(scanf("%d%d",&k,&m)!=EOF){
for(i=10;i>=1;i--)
scanf("%d",&c.M[10][i]);
for(i=1;i<=9;i++)
for(j=1;j<=10;j++)
c.M[i][j]=((j-1)==i);
c.l=10; c.r=10;
/*for(i=1;i<=10;i++){
for(j=1;j<=10;j++) printf("%d",c.M[i][j]);
cout<<endl;
}*/
for(i=0;i<=9;i++) res.M[i+1][1]=i;
res.l=10; res.r=1;
for(; k ; k >>= 1){
if(k & 1){
res * c;
}
c * c;
}
printf("%d\n",res.M[1][1]);
}
return 0;
}