A Simple Math Problem（矩阵快速幂）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1757

思路：矩阵快速幂模板题，不过因为刚刚入门矩阵快速幂，所以经常把数组f存反，导致本地错误一晚，差点心态爆炸……

代码实现如下：

#include <cstdio>
#include <cstring>

int k, m;
int a[10], f[10], mp[10][10];

void mulself(int a[10][10]) {
    int c[10][10];
    memset(c, 0, sizeof(c));
    for(int i = 0; i < 10; i++) {
        for(int j = 0; j < 10; j++) {
            for(int k = 0; k < 10; k++) {
                c[i][j] = (c[i][j] + (long long)a[i][k] * a[k][j]) % m;
            }
        }
    }
    memcpy(a, c, sizeof(c));
}

void mul(int f[10], int a[10][10]) {
    int c[10];
    memset(c, 0, sizeof(c));
    for(int i = 0; i < 10; i++) {
        for(int j = 0; j < 10; j++) {
            c[i] = (c[i] + (long long)f[j] * a[j][i] ) % m;
        }
    }
    memcpy(f, c, sizeof(c));
}

int main() {
    while(~scanf("%d%d", &k, &m)) {
        for(int i = 0; i < 10; i++) {
            scanf("%d", &a[i]);
        }
        for(int i = 0; i < 10; i++) {
            f[i] = 9 - i;
        }
        if(k <10) {
            printf("%d
", f[k]);
            continue;
        }
        memset(mp, 0, sizeof(mp));
        for(int i = 0; i < 10; i++) {
            mp[i][0] = a[i];
        }
        for(int i = 0; i < 9; i++) {
            mp[i][i + 1] = 1;
        }
        k = k - 9;
        for(; k; k >>= 1) {
            if(k & 1) mul(f, mp);
            mulself(mp);
        }
        printf("%d
", f[0]);
    }
    return 0;
}