【JZOJ 5215】【GDOI2018模拟7.9】组合数问题

Description

求：

$$(\sum_{i=1}^{\infty}C_{nk}^{ik+r})\mod P$$
$n<10^9,r<k<50,P<2^{30}-1$

Solution

指向【一大堆常犯的错误、提醒和公式】的第19条；
我们来考虑一下这个式子的意义：
在nk个物品中，选出x个物品，使得$x\mod k=r$
转化成这样以后就简单多了，
设DP $f_{i,j}$表示选了前i物品，余数为j的方案数，
暴力转移：$f_{i,j}=f_{i-1,j}+f_{i-1,((j-1+k)\mod k)}$

发现，这东西满足倍增的性质，就是可以从i转移到2i，
这样用矩阵乘法或倍增即可

复杂度：$O(k^3\log(n))$或$O(k^2\log(n))$

Code

#include <cstdio>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long LL;
const int N=55;
int m,mo,m1;
LL n,f[N],ans[N];
LL g[N];
LL ksm(LL q,LL w)
{
    LL ans=1;
    for(;w;w>>=1,q=q*q%mo)
    {
        if(w&1)ans=ans*q%mo;
    }
    return ans;
}
int main()
{
    int q,w;LL SR;
    scanf("%lld%d%d%d",&n,&mo,&m,&m1);
    n*=(LL)m;
    if(m==1)
    {
        printf("%lld",ksm(2,n));
        return 0;
    }
    ans[0]=1;
    f[1]=f[0]=1;
    for(;n;n>>=1)
    {
        if(n&1)
        {
            fo(i,0,m-1)fo(j,0,m-1)
                g[(i+j)%m]=(g[(i+j)%m]+f[i]*ans[j])%mo;
            fo(i,0,m-1)ans[i]=g[i],g[i]=0;
        }
        fo(i,0,m-1)fo(j,0,m-1)
            g[(i+j)%m]=(g[(i+j)%m]+f[i]*f[j]%mo)%mo;
        fo(i,0,m-1)f[i]=g[i],g[i]=0;
    }
    printf("%lld\n",ans[m1]);
    return 0;
}