Recurrences UVA - 10870 （斐波拉契的一般形式推广）

题意：f(n) = a1f(n−1) + a2f(n−2) + a3f(n−3) + ... + adf(n−d), 计算这个f(n)

最重要的是推出矩阵。

#include<cstdio>
#include<cstring>
#define ll long long
ll mod, d, n;
ll a[];
ll f[];
struct jz
{
    ll num[][];
    jz(){ memset(num, , sizeof(num)); }
    jz operator*(const jz&p)const
    {
        jz ans;
        for (int k = ; k < d; ++k)
        for (int i = ; i < d;++i)
        for (int j = ; j < d; ++j)
            ans.num[i][j] = (ans.num[i][j] + num[i][k] * p.num[k][j] % mod) % mod;
        return ans;
    }
}p;
jz POW(jz x, ll n)
{
    jz ans;
    for (int i = ; i < d; ++i)ans.num[i][i] = ;
    for (; n;n>>=, x=x*x)
    if (n & )ans = ans*x;
    return ans;
}
void init()
{
    for (int i = ; i < d; ++i)
        p.num[][i] = a[i];
    for (int i = ; i < d; ++i)
        p.num[i][i - ] = ;
}
int main()
{
    while (scanf("%lld%lld%lld", &d, &n, &mod) != EOF, d + n + mod)
    {
        for (int i = ; i < d; ++i)scanf("%lld", &a[i]);
        for (int i = ; i < d; ++i)scanf("%lld", &f[i]);
        if (n <= d){ printf("%lld\n", f[n - ]); }
        else
        {
            init();
            jz ans = POW(p, n - d);
            ll kk = ;
            for (int i = , j = d - ; i < d; ++i, --j)
            {
                kk = (kk + ans.num[][i] * f[j] % mod) % mod;
            }
            printf("%lld\n", kk);
        }
    }
    return ;
}