[bzoj4870][Shoi2017]组合数问题

来自FallDream的博客，未经允许，请勿转载，谢谢。

---

与sdoi2017那道题神似

用f[i]表示选出的数量膜k等于i的方案数，容易构造转移矩阵，复杂度k^3log(nk)

也可以构造一个生成函数 复杂度k^2log(nk) 

理论上可以优化到klognlogk 但常数巨大。

#include<iostream>
#include<cstdio>
#define MN 50
using namespace std;
inline int read()
{
    int x = 0 , f = 1; char ch = getchar();
    while(ch < '0' || ch > '9'){ if(ch == '-') f = -1;  ch = getchar();}
    while(ch >= '0' && ch <= '9'){x = x * 10 + ch - '0';ch = getchar();}
    return x * f;
}

int n,p,k,r,A[MN+5],B[MN+5],C[MN+5];

void Mul(int*a,int*b)
{
    for(int i=0;i<k;++i) if(a[i])
        for(int j=0;j<k;++j)
            C[(i+j)%k]=(C[(i+j)%k]+1LL*a[i]*b[j])%p; 
    for(int i=0;i<k;++i) a[i]=C[i],C[i]=0;
}

int main()
{
    n=read();p=read();k=read();r=read();
    A[0]=1;B[0]=1;++B[1%k];
    for(long long K=1LL*n*k;K;K>>=1,Mul(B,B))
        if(K&1) Mul(A,B); 
    printf("%d\n",A[r]);
    return 0;
}