[BZOJ5306] [HAOI2018]染色(容斥原理+NTT) 题面 一个长度为 n的序列, 每个位置都可以被染成 m种颜色中的某一种. 如果n个位置中恰好出现了 S次的颜色有 K种, 则小 C 会产生 \(W_k\)的愉悦度. 求对于所有可能的染色方案, 他能获得的愉悦度的和.答案对 1004535809 取模 分析 显然颜色数量不超过\(tot=\min(m,\frac{n}{S})\) 我们需要求出现了\(S\)次的颜色有\(i\)种的方案数.这个东西不太好求,考虑容斥,求出现了\(S…

[题解][HAOI2018]染色(NTT+容斥/二项式反演) 可以直接写出式子: \[ f(x)={m \choose x}n!{(\dfrac 1 {(Sx)!})}^x(m-x)^{n-Sx}\dfrac 1 {(n-Sx)!} \] \(f(x)\) 钦定有\(x\)种颜色出现了恰好\(S\)的方案 然后推一下恰好有\(x\)种颜色出现了恰好\(S\)次的方案\(g(x)\) .推导在下下面. 最后的答案是\(\sum w_i g(i)\) 推导: 显然颜色种类不会超过\(L=\lfloo…

容易想到枚举恰好出现S次的颜色有几种.如果固定至少有i种恰好出现S次,那么方案数是C(M,i)·C(N,i*S)·(M-i)N-i*S·(i*S)!/(S!)i,设为f(i). 于是考虑容斥,可得恰好i种的答案为Σ(-1)j-iC(j,i)·f(j) (j=i~min(M,⌊N/S⌋)).因为容斥是一个枚举子集的过程,在算至少i种的方案时,f(j)被计入了C(j,i)次. f显然可以通过预处理阶乘及其逆元线性地算出来.考虑怎么快速算后一部分.注意到模数,NTT没跑了.拆开组合数,可以发现是与j-…

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

bzoj luogu Description 给一个长度为\(n\)的序列染色,每个位置上可以染\(m\)种颜色.如果染色后出现了\(S\)次的颜色有\(k\)种,那么这次染色就可以获得\(w_k\)的收益. 求所有染色方案的收益之和膜\(1004535809\). sol 整行公式太大了放不下就只能用行内公式了qaq 首先设\(N=\min(m,\lfloor\frac ns\rfloor)\),这是出现了\(S\)次的颜色种数的上界. 设\(F(i)\)表示染色后出现了\(S\)次的颜色有\…

前置芝士 可重集排列 NTT 前置定义 \[\begin{aligned}\\ f_i=C_m^i\cdot \frac{n!}{(S!)^i(n-iS)!}\cdot (m-i)^{n-iS}\\ ans_i=\sum\limits_{j=i}^lim (-1)^{j-i}C_j^i f_j\\ \end{aligned}\] 理解:\(m\)种颜色选i种恰好出现\(S\)次,可重全排列,剩余块染色,不过这样有可能会出现剩余块种有恰好出现\(S\)次的情况,所以容斥一下 \(C_j^i\):\…

https://www.cnblogs.com/zhoushuyu/p/9138251.html 注意如果一开始F(i)中内层式子中j枚举的是除前i种颜色之外还有几种出现S次的颜色,那么后面式子就会难推很多. #include<cstdio> #include<algorithm> #define rep(i,l,r) for (int i=(l); i<=(r); i++) using namespace std; ,M=,mod=; int n,m,s,ans,w[N],…

BZOJ5306 [Haoi2018]染色 Solution xzz的博客 代码实现 #include<stdio.h> #include<stdlib.h> #include<string.h> #include<math.h> #include<algorithm> #include<queue> #include<set> #include<map> #include<iostream> us…

[BZOJ5306]染色(NTT) 题面 BZOJ 洛谷 题解 我们只需要考虑每一个\(W[i]\)的贡献就好了 令\(lim=min(M,\frac{N}{S})\) 那么,开始考虑每一个\(W[i]\)的贡献 \[\sum_{k=0}^{lim}W[k]C_M^kC_N^{kS}\frac{(kS)!}{(S!)^k}\times Others\] \(Others\)是其他的东西,先考虑前面这堆东西的意义. 我们枚举恰好出现了\(S\)次的颜色个数\(k\),那么,选定这些颜色的方案数 首…

BZOJ 5306 [HAOI2018] 染色 首先,求出$N$个位置,出现次数恰好为$S$的颜色至少有$K$种. 方案数显然为$a_i=\frac{n!\times (m-i)^{m-i\times s}}{(m-K)!\times (s!)^K}\times C(m,K)$ 然后二项式反演一下,得到恰好的数量:$ans_i=\sum\limits_{j=i}^n (-1)^{j-i}\times a_i\times C(j,i)$ 然后展开一下就可以得到两个多项式:$A_i=\frac{m!…