题目大意
有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;
}
}