BZOJ4161 常系数齐次线性递推

问了数竞的毛毛搞了一番也没太明白，好在代码蛮好写先记下吧。

#include<bits/stdc++.h>
using namespace std;
const int N=4005,mod=1e9+7;
int n,k,c[N],b[N],a[N],f[N],tmp[N],ans;
inline void qmul(int *x,int *y)
{
    for(int i=0;i<k*2;++i)tmp[i]=0;
    for(int i=0;i<k;++i)
    for(int j=0;j<k;++j)
    tmp[i+j]=(tmp[i+j]+1ll*x[i]*y[j]%mod)%mod;
    for(int i=k*2-2;i>=k;--i)
    for(int j=1;j<=k;++j)
    tmp[i-j]=(tmp[i-j]+1ll*tmp[i]*a[j]%mod)%mod;
    for(int i=0;i<k;++i)x[i]=tmp[i];
    return;
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=k;++i)scanf("%d",&a[i]),a[i]=(a[i]%mod+mod)%mod;
    for(int i=0;i<k;++i)scanf("%d",&f[i]),f[i]=(f[i]%mod+mod)%mod;
    c[1]=b[0]=1;
    if(n<k){
        printf("%d
",f[n]);
        return 0;
    }
    while(n)
    {
        if(n&1)qmul(b,c);
        qmul(c,c);n>>=1;
    }
    for(int i=0;i<k;++i)ans=(ans+1ll*f[i]*b[i]%mod)%mod;
    printf("%d
",ans);
    return 0;
}