卧薪尝胆,厚积薄发。
GDOI模拟 朝圣
Date: Thu Nov 29 10:26:32 CST 2018
In Category:
NoCategory
Description:
对于
$m\in[0,M]$
,求出在有
$n$
个苹果
$m$
个梨时选出
$k$
个水果恰好有质数个苹果的方案数,苹果和梨各不相同。
$1\leqslant n\leqslant10^5$
Solution:
先列式子:
$$
ans[m]=\frac{\begin{align}\sum_{i=1}^k[i\ is\ prime]\times C_n^i\times C_m^{k-i}\end{align}}{C_{n+m}^k}
$$
考虑怎么把它推成卷积的形式,先把式子中只与
$i$
有关的拿出来,即:
$$
\sum_{i=1}^k[i\ is\ prime]C_n^i\frac{m!}{(k-i)!}\times\frac{1}{(m-k+i)!}
$$
设
$a[i]=[i\ is\ prime]\times C^i_n\times\frac {M!}{(k-i)!},b[i]=\frac{1}{(-i)!}$
那么:
$$
ans[m]=\sum_{i=0}^ka[i]\times b[k-m-i]
$$
令
$C=A\times B$
,那么
$ans[m]=C[k-m]$
,可以用
$NTT$
求解。
Code:
懒得写代码了
In tag:
数学-多项式-快速数论变换
Copyright © 2020
wjh15101051
ღゝ◡╹)ノ♡