[HAOI2018] 染色二项式反演

最新推荐文章于 2024-11-11 22:10:43 发布

DrGilbert

最新推荐文章于 2024-11-11 22:10:43 发布

阅读量188

点赞数

分类专栏：组合计数文章标签： acm竞赛算法二项式二项式反演

本文链接：https://blog.csdn.net/oampamp1/article/details/110404570

版权

这篇博客介绍了如何运用二项式反演解决一个ACM竞赛中的染色问题，目标是计算所有可能染色方案中恰好出现特定次数颜色的愉悦度之和。通过分析题面，将'恰好出现S次'转化为'至少S次'或'至多S次'，利用二项式反演公式推导出方案数，并用NTT优化算法实现O(N+nlogn)的复杂度。

摘要由CSDN通过智能技术生成

原题题面

为了报答小 C 的苹果, 小 G 打算送给热爱美术的小 C 一块画布, 这块画布可以抽象为一个长度为 $N$ 的序列, 每个位置都可以被染成 $M$ 种颜色中的某一种.
然而小 C 只关心序列的 $N$ 个位置中出现次数恰好为 $S$ 的颜色种数, 如果恰好出现了 $S$ 次的颜色有 $K$ 种, 则小 C 会产生 $W_k$ 的愉悦度.
小 C 希望知道对于所有可能的染色方案, 他能获得的愉悦度的和对 $1004535809$ 取模的结果是多少.

输入格式

从标准输入读入数据. 第一行三个整数 $N(1\leq N \leq 10^7),M(1\leq M \leq 10^5),S(1\leq S \leq 150)$ .
接下来一行 $M + 1$ 个整数, 第 $i$ 个数表示 $W_{i-1}$ .

输出格式

输出到标准输出中. 输出一个整数表示答案.

输入样例

8 8 3
3999 8477 9694 8454 3308 8961 3018 2255 4910

输出样例

524070430

题面分析

注意到题面是是恰好出现S次，因此考虑利用二项式反演把问题将“恰好出现S次”转化为“至少S次”或者“至多S次”。
我们考虑至少S次。
已知颜色出现种类的上限 $limit=min(M,\frac{N}{S})$
记恰好出现S次的颜色种类至少有 $i$ 种的方案数为 $f (i)$ ，
我们将 $n$ 分为 $i + 1$ 个位置，则方案数为
$f(i)=C_{M}^{i}*\frac{N!}{(S!)^i(N-iS)!}*(M-i)^{N-iS}$