bzoj 4161 Shlw loves matrixI【常系数线性齐次递推】

并不会递推,不过板子挺好背的,只要是类似的递推都能用,但是注意c数组不能使负数
如果除了递推还有常数项的话,就用f[i]-f[i-1]的方式消掉常数项(然后多一个f[i-1]的项)

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
const int N=4005,mod=1000000007;
int n,m,a[N],ans[N],c[N],d[N],f[N],an;
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void jia(int &x,int y)
{
    x+=y;
    x>=mod?x-=mod:0;
}
int ksm(int a,int b)
{
    int r=1;
    while(b)
    {
        if(b&1)
            r=1ll*r*a%mod;
        a=1ll*a*a%mod;
        b>>=1;
    }
    return r;
}
void mul(int a[],int b[])
{
    memset(d,0,sizeof(d));
    for(int i=0;i<=m;i++)
        for(int j=0;j<=m;j++)
            jia(d[i+j],1ll*a[i]*b[j]%mod);
    for(int i=2*m;i>=m;i--)
    {
        for(int j=0;j<m;j++)
            jia(d[i-m+j],1ll*d[i]*c[m-j]%mod);
        d[i]=0;
    }
    for(int i=0;i<=2*m;i++)
        a[i]=d[i];
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=m;i++)
        c[i]=(read()+mod)%mod;
    for(int i=0;i<m;i++)
        f[i]=(read()+mod)%mod;
    a[1]=1,ans[0]=1;
    while(n)
    {
        if(n&1)
            mul(ans,a);
        mul(a,a);
        n>>=1;
    }
    for(int i=0;i<m;i++)
        jia(an,1ll*f[i]*ans[i]%mod);
    printf("%lld\n",an);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/lokiii/p/10275892.html