A Simple Math Problem（矩阵快速幂）

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

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

代码实现如下：

 #include <cstdio>
 #include <cstring>

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

 void mulself(int a[][]) {
     int c[][];
     memset(c, , sizeof(c));
     for(int i = ; i < ; i++) {
         for(int j = ; j < ; j++) {
             for(int k = ; k < ; 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[], int a[][]) {
     int c[];
     memset(c, , sizeof(c));
     for(int i = ; i < ; i++) {
         for(int j = ; j < ; 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 = ; i < ; i++) {
             scanf("%d", &a[i]);
         }
         for(int i = ; i < ; i++) {
             f[i] =  - i;
         }
         if(k <) {
             printf("%d\n", f[k]);
             continue;
         }
         memset(mp, , sizeof(mp));
         for(int i = ; i < ; i++) {
             mp[i][] = a[i];
         }
         for(int i = ; i < ; i++) {
             mp[i][i + ] = ;
         }
         k = k - ;
         for(; k; k >>= ) {
             if(k & ) mul(f, mp);
             mulself(mp);
         }
         printf("%d\n", f[]);
     }
     return ;
 }