好坑啊不开心……

  其实这题的想法还是比较简单粗暴的。题目明示恰好xxx,显然排除斜率二分这个玩意儿,那么不就只剩下容斥了嘛……

  令 \(A_{x}\) 为恰好出现了 \(S\) 次的至少有 \(x\) 种的方案数, \({B_{x}}\) 为恰好出现了\(S\) 次的颜色恰好 \(x\) 种的方案数。\(A_{x}\) 可以 \(O(1)\) 求得,\(A_{x} = \frac{\binom{n}{S * x} * (m - x) ^ {n - S * x} * (S * i)!}{(S!)^{x}}\)。(为了方便描述,我们在下面所说的颜色均为满足恰好出现 \(S\) 次的颜色)。

那么,一个恰好有 \(x\) 种颜色的方案对答案的贡献应该为

(我们先只考虑颜色种数恰好为 \(k\) 的)

\(g_{x} = [x = k] = \sum_{i = 0}^{x}\binom{x}{i}f_{i}\)

(\(f_{i}\) 为容斥系数)

二项式反演,得

\(f_{x} = \binom{k}{x} * (-1) ^ {k - x}\)

所以令 \(B_{x}\) 为恰好 \(x\) 种颜色的方案数,有:

\(B_{x} = \sum_{i = 0}^{m}\binom{m}{i}*A_{i}*\binom{i}{x}*(-1)^{i - x}\)

由于有 \(m\) 个\(B_{x}\) 要求值,且明显的给了一个NTT模数,

考虑转化成卷积的形式:

(把与各种变量相关的尽量整理到一起)

得到:\(B_{x} = \frac{(-1)^{-x}}{x!}\sum_{i = 0}^{m}\frac{\binom{m}{i}*(-1)^{i}}{(i - x)!}\)

前面的是个常数,后面的是一个卷积(把 \(i - x\) 先 \(+ m\) 再反转(防止爆负))……上NTT就好了。

  但这还没完!预处理逆元、阶乘逆元、阶乘的话会容易TLE!所以应该快速幂暴力处理逆元……(;′⌒`)

#include <bits/stdc++.h>
using namespace std;
#define maxn 10000005
#define mod 1004535809
#define int long long
int n, m, M, S, W[maxn], G[maxn], a[maxn], b[maxn], c[maxn];
int ans, K, fac[maxn], A[maxn], B[maxn];
int lim = , len, rev[maxn]; int read()
{
int x = , k = ;
char c; c = getchar();
while(c < '' || c > '') { if(c == '-') k = -; c = getchar(); }
while(c >= '' && c <= '') x = x * + c - '', c = getchar();
return x * k;
} int Up(int x) { if(x >= mod) x -= mod; return x; }
int Down(int x) { if(x < ) x += mod; return x; }
int Mod(int x) { x %= mod; if(x < ) x += mod; return x; }
int Qpow(int x, int timer)
{
int base = ;
for(; timer; timer >>= , x = x * x % mod)
if(timer & ) base = x * base % mod;
return base;
}
#define inv(x) Qpow(x, mod - 2) int C(int n, int m)
{
if(n < m || n < || m < ) return ;
int t = fac[n] * inv(fac[m]) % mod * inv(fac[n - m]) % mod;
return t;
} void init()
{
fac[] = ;
int t = max(n, m);
for(int i = ; i <= t; i ++) fac[i] = fac[i - ] * i % mod;
} void Pre()
{
for(int i = ; i <= m; i ++)
{
if(S * i > n) break;
int t = S * i, t1 = C(n, t) * fac[t] % mod;
A[i] = t1 * Qpow(inv(fac[S]), i) % mod * Qpow(m - i, n - t) % mod;
}
for(int i = ; i <= m; i ++)
a[i] = C(m, i) * A[i] % mod * fac[i] % mod * Qpow(-, i) % mod;
for(int i = -m; i <= m; i ++) c[i + K] = i < ? : inv(fac[i]);
for(int i = ; i <= M; i ++) b[i] = c[M - i];
} void NTT(int *A, int opt)
{
int t = opt < ? Qpow(, mod - ) : ;
for(int i = ; i < lim; i ++)
if(i < rev[i]) swap(A[i], A[rev[i]]); for(int l = ; l < lim; l <<= )
{
int g = Qpow(t, (mod - ) / (l << ));
for(int i = ; i < l; i ++) G[i] = 1ll * G[i - ] * g % mod;
for(int i = ; i < lim; i += (l << ))
for(int j = i; j <= i + l - ; j ++)
{
int x = A[j], y = 1ll * G[j - i] * A[j + l] % mod;
A[j] = Up(x + y); A[j + l] = Down(x - y);
}
}
} signed main()
{
n = read(), m = read(), S = read(); K = m, M = m * + ;
for(int i = ; i <= m; i ++) W[i] = read();
init(), Pre(); G[] = ;
while(lim <= M + m) lim <<= , len ++;
for(int i = ; i < lim; i ++)
rev[i] = (rev[i >> ] >> ) | ((i & ) << ((len - )));
NTT(a, ), NTT(b, );
for(int i = ; i < lim; i ++) a[i] = a[i] * b[i] % mod;
NTT(a, -); int inv1 = Qpow(lim, mod - );
for(int i = ; i <= m; i ++)
{
int t = ((i & ) ? - : ) * inv(fac[i]) % mod * W[i] % mod;
ans = Mod(ans + a[i + m + ] * inv1 % mod * t % mod);
}
printf("%lld\n", ans);
return ;
}

【题解】HAOI2018染色的更多相关文章

  1. 【题解】[HAOI2018]染色(NTT+容斥/二项式反演)

    [题解][HAOI2018]染色(NTT+容斥/二项式反演) 可以直接写出式子: \[ f(x)={m \choose x}n!{(\dfrac 1 {(Sx)!})}^x(m-x)^{n-Sx}\d ...

  2. [洛谷P4491] [HAOI2018]染色

    洛谷题目链接:[HAOI2018]染色 题目背景 HAOI2018 Round2 第二题 题目描述 为了报答小 C 的苹果, 小 G 打算送给热爱美术的小 C 一块画布, 这块画布可 以抽象为一个长度 ...

  3. 【LG4491】[HAOI2018]染色

    [LG4491][HAOI2018]染色 题面 洛谷 题解 颜色的数量不超过\(lim=min(m,\frac nS)\) 考虑容斥,计算恰好出现\(S\)次的颜色至少\(i\)种的方案数\(f[i] ...

  4. 题解-HAOI2018全套

    去冬令营转了一圈发现自己比别人差根源在于刷题少,见过的套路少(>ω<) 于是闲来无事把历年省选题做了一些 链接放的都是洛谷的,bz偷懒放的也是链接 AM.T1 奇怪的背包 Problem ...

  5. BZOJ 5306 [HAOI2018] 染色

    BZOJ 5306 [HAOI2018] 染色 首先,求出$N$个位置,出现次数恰好为$S$的颜色至少有$K$种. 方案数显然为$a_i=\frac{n!\times (m-i)^{m-i\times ...

  6. 【BZOJ5306】 [Haoi2018]染色

    BZOJ5306 [Haoi2018]染色 Solution xzz的博客 代码实现 #include<stdio.h> #include<stdlib.h> #include ...

  7. [BZOJ5306] [HAOI2018]染色(容斥原理+NTT)

    [BZOJ5306] [HAOI2018]染色(容斥原理+NTT) 题面 一个长度为 n的序列, 每个位置都可以被染成 m种颜色中的某一种. 如果n个位置中恰好出现了 S次的颜色有 K种, 则小 C ...

  8. [题解 LuoguP4491 [HAOI2018]染色

    传送门 神仙计数题 Orz 先令\(F[k]\)表示出现次数恰好为\(S\)次的颜色恰好有\(k\)中的方案数,那么 \[Ans=\sum\limits_{i=0}^mW_iF[i]\] 怎么求\(F ...

  9. 【BZOJ5306】[HAOI2018]染色(NTT)

    [BZOJ5306]染色(NTT) 题面 BZOJ 洛谷 题解 我们只需要考虑每一个\(W[i]\)的贡献就好了 令\(lim=min(M,\frac{N}{S})\) 那么,开始考虑每一个\(W[i ...

随机推荐

  1. linux(模糊批量删除文件)删除指定文件夹中某个文件除外的其他文件

    # shopt -s extglob# rm -fr !(file1)如果是多个要排除的,可以这样:# rm -rf !(file1|file2)首先科普下shopt -s extglobBash S ...

  2. maven scope属性值设置含义

    1.枚举各个属性值的含义 compile,缺省值,适用于所有阶段,会打包进项目. provided,类似compile,期望JDK.容器或使用者会提供这个依赖. runtime,只在运行时使用,如JD ...

  3. Oracle扩展包(pipe,alert,job,scheduler)

    --定义包中函数的纯度级别 create or replace package purityTest is type dept_typ is table of dept%rowtype index b ...

  4. 【Unity Shader】渲染管线

    流程概述 应用程序阶段 应用程序阶段,使用高级编程语言(C.C++.JAVA 等)进行开发,主要和CPU.内存打交道,诸如碰撞检测.场景图建立.空间八叉树更新.视锥裁剪等经典算法都在此阶段执行.在该阶 ...

  5. yocto-sumo源码解析(六): setup_bitbake

    1. 创造日志handler: 在status_only模式,不需要日志以及UI # Ensure logging messages get sent to the UI as events hand ...

  6. PHP has encountered an Access Violation at 01F4A622解决方法

    php搭建的网站出现以下问题的解决方法分享: Z-blog,DedeCMS,Dsicuz!,PhpWind,PhpCMS,帝国CMS等都有可能出现php访问冲突问题. 今天访问网站发现出现了一个错误& ...

  7. 【视频编解码·学习笔记】13. 提取PPS信息程序

    PPS结构解析 与之前解析SPS方式类似 一.定义PPS类: 在3.NAL Unit目录下,新建PicParamSet.cpp和PicParamSet.h,在这两个文件中写入类的定义和函数实现. 类定 ...

  8. 基于NABCD评论作业-王者荣耀交流协会PSP DAILY

    一.根据(不限于)NABCD评论作品的选题   N(Need,需求):在我知道PSP DAILY这款软件的时候,就认为这款软件对于学习软件工程课的学生来说有很大的需要.对于需求来说,软件工程课程中的学 ...

  9. iOS开发学习-cocoapods的配置安装

    安装coacoapods步骤: 在终端输入如下命令,升级ruby版本: sudo gem update —system 出现这个,标志着安装成功. 完成之后,再输入如下命令: gem sources ...

  10. Task 4.5 求二维数组中的最大连通子数组之和

    任务:输入一个二维整形数组,数组里有正数也有负数. 求所有子数组的和的最大值.要求时间复杂度为O(n). 1.设计思想:因为用之前的解决子数组最大和的问题的思路一直没能解决这个问题,后来看到同学使用将 ...