HDU1757：A Simple Math Problem（矩阵快速幂）

#include <iostream>
#include <string>
#include <stdlib.h>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct ma
{
    int a[10][10];
} init,res;
int K;
int mod,b[10],f[10];
ma Mul(ma x,ma y)
{
    ma tmp;
    for(int i=0; i<10; i++)
        for(int j=0; j<10; j++)
        {
            tmp.a[i][j]=0;
            for(int k=0; k<10; k++)
                tmp.a[i][j]=(tmp.a[i][j]+x.a[i][k]*y.a[k][j])%mod;
        }
    return tmp;
}
ma Pow(ma x,int K)
{
    ma tmp;
    for(int i=0; i<10; i++)
    {
        for(int j=0; j<10; j++)
            tmp.a[i][j]=(i==j);
    }
    while(K!=0)
    {
        if(K&1)
            tmp=Mul(tmp,x);
        K>>=1;
        x=Mul(x,x);
    }
    return tmp;
}
int main()
{
    while(scanf("%d%d",&K,&mod)!=EOF)
    {
        for(int i=0; i<=9; i++)
        {
            scanf("%d",&init.a[9][9-i]);
        }
        if(K<=9)
        {
            printf("%d\n",K);
            continue;
        }
        for(int i=0; i<10; i++)
            f[i]=i;
        for(int i=0; i<=8; i++)
        {
            for(int j=0; j<=9; j++)
                init.a[i][j]=(i==j-1);
        }
        res=Pow(init,K-9);
        int ans=0;
        for(int j=0; j<10; j++)
        {
            ans=(ans+(res.a[9][j])*f[j])%mod;
        }
        printf("%d\n",ans);

    }
    return 0;
}