hdu 1757 A Simple Math Problem

很水的一道矩阵快速幂。

#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;
int g[10],k,mod;
struct rec{
    ll v[10][10];
    rec(){memset(v,0,sizeof v);}
};
rec mul(rec a,rec b){
    rec ans;
    ll sum;
    for(int k=0;k<10;k++){
       for(int i=0;i<10;i++){
          for(int j=0;j<10;j++){
              sum=a.v[i][k]*b.v[k][j]%mod;
              ans.v[i][j]=(ans.v[i][j]+sum)%mod;
          }
       }
    }
    return ans;
}
rec pow(rec a,int n){
    if(n==1) return a;
    rec ans=pow(a,n/2);
    ans=mul(ans,ans);
    if(n&1) return mul(ans,a);
    else return ans;
}
int f(rec a){
    int ret=0;
    for(int i=0;i<10;i++){
        ret=(ret+a.v[0][i]*(9-i)%mod)%mod;
    }
    return ret;
}
int main(){
    while(~scanf("%d%d",&k,&mod)){
        if(k<10) {printf("%d
",k);continue;}
        for(int i=0;i<10;i++) scanf("%d",&g[i]);
        rec a;
        for(int i=0;i<10;i++) a.v[0][i]=g[i];
        for(int i=1;i<10;i++) a.v[i][i-1]=1;
        a=pow(a,k-9);
        printf("%d
",f(a));
    }
    return 0;
}