快速幂&矩阵快速幂

#include <iostream>
#include <string.h>
#include <stdio.h>
/*
int pow(int a, int b)
{
    long long ans = 1, base = a;
    while(b != 0)
    {
        if(b & 1)
        {
            ans *= base;
        }
        base *= base;
        b>>=1;
    }
    return ans;
}
快速幂
*/
const int n = 10;
int k, mod, arr[n], f[n];

struct matrix
{
    long long mat[n][n];
    matrix operator*(const matrix &m)const
    {
        matrix tmp;
        for(int i = 0; i < n; i++)
        {
            for(int j = 0; j < n; j++)
            {
                tmp.mat[i][j] = 0;
                for(int k = 0; k < n; k++)
                {
                    tmp.mat[i][j] += mat[i][k] * m.mat[k][j] % mod;
                    tmp.mat[i][j] %= mod;
                }
            }
        }
        return tmp;
    }
};

long long pow(matrix &m, int k){
matrix ans;
memset(ans.mat, 0, sizeof(ans.mat));
for(int i = 0; i < n; i++)
    {
        ans.mat[i][i] = 1;
    }
    k -= 9;
    while(k)
    {
        if(k & 1)
        {
            ans = ans * m;
        }
        k >>= 1;
        m = m * m;
    }
    long long sum = 0;
    for(int i = 0; i < n; i++)
    {
        sum += ans.mat[0][i]*f[n - i - 1] % mod;
        sum %= mod;
    }
    return sum;
}

void init(matrix &m)
{
    memset(m.mat, 0, sizeof(m.mat));
    for(int i = 0; i < n; i++)
    {
        m.mat[0][i] = arr[i];
    }
    for(int i = 0; i < n - 1; i++)
    {
        m.mat[i + 1][i] = 1;
    }
    for(int i = 0; i < n; i++)
    {
        f[i] = i;
    }
}

int main()
{
    matrix m;
    while(scanf("%d%d", &k, &mod) != EOF)
    {
        for(int i = 0; i < n; i++)
        {
            scanf("%d", &arr[i]);
        }
        init(m);
        if(k < 10)
        {
            printf("%d\n", k % mod);
        }
        else{
            printf("%I64d\n", pow(m, k));
        }
    }
    return 0;
}