首页 > 代码库 > BZOJ4555求和(cdq分治+NTT)

BZOJ4555求和(cdq分治+NTT)

题意:

技术分享

输出f(n)对998244353(7 × 17 × 223 + 1)取模的结果。1 ≤ n ≤ 100000

其中S(i,j)是第二类Stirling数,即有i个球,丢到j个盒子中,要求盒子不为空的方案总数

S(i,j)=S(i-1,j-1)+j*S(i-1,j) (前面一项表示第i个球单独放到一个盒子中,后面一项表示放到前面j个盒子中的某一个)

分析:

首先这个n不是丧心病狂的大,所以感觉可以求i=1时的结果,i=2时的结果,i=3时的结果……,于是可以不看第一个Σ

我们考虑后面的这项

技术分享

这个表示有n个球,要丢到i个盒子中,这些盒子是有序的,并且球在盒子中有两种状态

 换个方式枚举,枚举最后一个盒子中球的个数,那么有

技术分享

右边这个有卷积的感觉了,再来弄一下,将组合公式带进去

技术分享

令G(n)=g(n)/n!

那么有技术分享

这个inv(n-i)表示模意义下(n-i)!的逆元,这个预处理很简单

问题是这个并不是很简单的卷积,因为卷积是A(n)=ΣB(i)*C(n-i),但这里A=C

 这里可以采取cdq分治的思想

 我们要求G(l..r)

第一步:求G(l..mid)

第二步:求G(l..mid)对G(mid+1,r)的影响

第三步:求G(mid+1,r)

具体说下第二步

这时候G(l,mid)都已经确定了,它们对G(mid+1)的贡献是容易发现就是G(0,mid-l)和inv(0,r-l)的卷积

所以就是cdq分治过程中进行FFT

但因为这题数字很大,所以用FFT会出现精度上的问题

而且这题模数那么特殊,所以直接上NTT……

T(n)=T(n/2)+O(nlogn)

这个复杂度应该介于O(nlogn)和O(nlog^2n)之间吧……

BZOJ4555求和(cdq分治+NTT)