[JZOJ5550]卡常题

题目大意

有k堆花，每堆都一样，一堆有m种花，每种有e[i]个，输入还会有个没用的c[i]表示颜色。设n=sum{e[i]}。
现在你从第0堆轮流采花，每次可以采任意种任意个，采了k-1继续采0，如果一堆的花全部采完，就停下来。现在问对于每堆花，在这里结束的采花方案。不同方案定义为存在某次采花的某种花采的数量不同。对985661441取模。每个测试点有T组数据。
n<=1e5,T<=200，m,k<=10

题解

只考虑一堆。设f[i]为第i天恰好采完的方案数，发现一个性质，第i天没有采完的方案数恰好是f[i+1]。
考虑怎么算f。一个好思路是对每种颜色分开使用隔板原理再累乘起来，其中隔板中间可以没有东西。但这样会出现一天什么都没有选的不合法情况，考虑容斥掉。那么得出式子。

然后你用ntt优化一下就行了…

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
typedef long long ll;
typedef double db;
#define fo(i,j,k) for(i=j;i<=k;i++)
#define fd(i,j,k) for(i=j;i>=k;i--)
#define cmax(a,b) (a=(a>b)?a:b)
#define cmin(a,b) (a=(a<b)?a:b)
const int N=4e5+5,mo=985661441,rt=3;
int fac[N],rev[N],w[20],rw[20],t[N],f[N],b[N],prod1[12],prod2[12],prt[12],e[12];
int Len,Ci,len,ci,m,K,n,j,revl,i,s,T,x,g[N];
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 c(int n,int m)
{
    return 1ll*fac[m]*rev[n]%mo*rev[m-n]%mo;
}
int ksm(int x,int y)
{
    int ret=1;
    while (y)
    {
        if (y&1) ret=1ll*ret*x%mo;
        y>>=1;
        x=1ll*x*x%mo;
    }
    return ret;
}
void predo(int n)
{
    int i,x;
    fac[0]=1;
    fo(i,1,n) fac[i]=1ll*fac[i-1]*i%mo;
    rev[n]=ksm(fac[n],mo-2);
    fd(i,n,1) rev[i-1]=1ll*rev[i]*i%mo;
    fo(i,0,Ci)
    {
        w[i]=ksm(rt,mo-1>>i);
        rw[i]=ksm(w[i],mo-2);
    }
}
void dft(int *a,int n,int sig)
{
    int i,j,k,tw,siz,half,W,v,tc;
    fo(i,0,n-1)
    {
        k=0;
        fo(j,0,ci-1) if (i&(1<<j))k+=1<<ci-1-j;
        t[k]=a[i];
    }
    for(siz=2,tc=1;siz<=n;siz<<=1,tc++)
    {
        half=siz>>1;
        if (sig==1) W=w[tc];else W=rw[tc];
        for(i=0;i<len;i+=siz)
        {
            tw=1;
            fo(j,i,i+half-1)
            {
                v=1ll*t[j+half]*tw%mo;
                t[j+half]=(t[j]-v<0)?t[j]-v+mo:t[j]-v;
                t[j]=(t[j]+v>=mo)?t[j]+v-mo:t[j]+v;
                tw=1ll*tw*W%mo;
            }
        }
    }
    fo(i,0,n-1) a[i]=t[i];
    if (sig==-1) 
    {
        revl=ksm(n,mo-2);
        fo(i,0,n-1) a[i]=1ll*a[i]*revl%mo;
    }
}
int main()
{
    freopen("t3.in","r",stdin);
    //freopen("t3.out","w",stdout);
    Len=1;
    while (Len<2e5+1) Len*=2,Ci++;
    predo(Len);
    while (scanf("%d %d",&m,&K)!=EOF)
    {
        n=0;
        fo(i,1,m)
        {
            j=read();
            e[i]=read();
            n+=e[i];
        }
        fo(j,1,n)//b[j]
        {
            b[j]=1;
            fo(i,1,m)
                b[j]=1ll*b[j]*c(j-1,e[i]+j-1)%mo;
            b[j]=1ll*b[j]*rev[j]%mo;
        }
        s=1;
        fo(i,0,n)
        {
            f[i]=(s*rev[i]+mo)%mo;
            s=-s;
        }
        len=1;ci=0;
        while (len<2*n+1) len*=2,ci++;
        dft(f,len,1);
        dft(b,len,1);
        fo(i,0,len-1) f[i]=1ll*f[i]*b[i]%mo;
        dft(f,len,-1);//*revl
        fo(i,0,len-1) f[i]=(1ll*f[i]*fac[i]%mo+mo)%mo;
        fo(i,n+1,len-1) f[i]=0;
        fo(i,0,K-1) prt[i]=0;
        prod1[0]=prod2[0]=1;
        fo(i,1,n)
        {
            fo(j,1,K) 
            {
                prod1[j]=1ll*prod1[j-1]*f[i+1]%mo;
                prod2[j]=1ll*prod2[j-1]*f[i]%mo;
                if (i==1) prod2[j]=f[i];
            }
            fo(j,0,K-1)
                 prt[j]=(prt[j]+1ll*prod1[j]*prod2[K-j])%mo;
        }
        printf("Case #%d:",++T);
        fo(i,0,K-1) printf(" %d",prt[i]);
        printf("\n");
        fo(i,0,len-1) b[i]=f[i]=0;
    }
}