Solution -「HAOI 2018」「洛谷 P4491」染色
\(\mathcal{Description}\)
Link.
用 \(m\) 种颜色为长为 \(n\) 的序列染色,每个位置一种颜色。对于一种染色方案,其价值为 \(w(\text{出现恰 }s\text{ 次的颜色种数})\)(\(w(0..m)\) 给定),求所有染色方案的价值和。
\(n\le10^7\),\(m\le10^5\),答案对 \(p=1004535809=479\times2^{21}+1\) 取模。
\(\mathcal{Solution}\)
记 \(l=\min\{m,\lfloor\frac{n}{s}\rfloor\}\),显然只能对于 \(i=0..l\),分别算出 \(w(i)\) 的贡献次数。考虑容斥,令 \(f(i)\) 表示至少 \(i\) 种颜色恰好出现 \(s\) 次的染色方案数。那么:
\]
即:选出 \(i\) 种颜色钦定恰好出现 \(s\) 次;多重集排列安排已被染色的 \(is\) 个位置和剩下的 \((n-is)\) 个位置;为 \((n-is)\) 个位置任选未使用的颜色。
此后,令 \(g(i)\) 表示恰好 \(i\) 种颜色恰好出现 \(s\) 次的染色方案数,容斥:
g(i)&=\sum_{j=i}^l(-1)^{l-i}\binom{j}if_j\\
&=\sum_{j=i}^l(-1)^{l-j}\frac{j!}{i!(j-i)!}f_j\\
&=\frac{1}{i!}\sum_{j=i}^l\frac{(-1)^{l-j}}{(j-i)!}(j!f_j)
\end{aligned}
\]
很俗套 trick,翻转乘积式任意一项就能写成卷积形式,NTT 求出 \(g(i)\) 即可。
复杂度 \(\mathcal O(n+l\log l)\)。
\(\mathcal{Code}\)
/* Clearink */
#include <cstdio>
#define rep( i, l, r ) for ( int i = l, rpbound##i = r; i <= rpbound##i; ++i )
#define per( i, r, l ) for ( int i = r, rpbound##i = l; i >= rpbound##i; --i )
const int MAXN = 1e7, MAXM = 1e5, MOD = 1004535809, G = 3, MAXLEN = 1 << 18;
int n, m, s, w[MAXM + 5], f[MAXLEN + 5], g[MAXLEN + 5], rev[MAXLEN + 5];
int fac[MAXN + 5], ifac[MAXN + 5]; // warning: 77MB.
inline char fgc () {
static char buf[1 << 17], *p = buf, *q = buf;
return p == q && ( q = buf + fread ( p = buf, 1, 1 << 17, stdin ), p == q )
? EOF : *p++;
}
inline int rint () {
int x = 0; char s = fgc ();
for ( ; s < '0' || '9' < s; s = fgc () );
for ( ; '0' <= s && s <= '9'; s = fgc () ) x = x * 10 + ( s ^ '0' );
return x;
}
inline void iswp ( int& a, int& b ) { a ^= b ^= a ^= b; }
inline int imin ( const int a, const int b ) { return a < b ? a : b; }
inline int imax ( const int a, const int b ) { return a < b ? b : a; }
inline int mul ( const long long a, const int b ) { return a * b % MOD; }
inline int sub ( int a, const int b ) { return ( a -= b ) < 0 ? a + MOD : a; }
inline int add ( int a, const int b ) { return ( a += b ) < MOD ? a : a - MOD; }
inline int mpow ( int a, int b ) {
int ret = 1;
for ( ; b; a = mul ( a, a ), b >>= 1 ) ret = mul ( ret, b & 1 ? a : 1 );
return ret;
}
inline int inv ( const int n ) { return mpow ( n, MOD - 2 ); }
inline void init ( const int n ) {
fac[0] = 1;
rep ( i, 1, n ) fac[i] = mul ( i, fac[i - 1] );
ifac[n] = inv ( fac[n] );
per ( i, n - 1, 0 ) ifac[i] = mul ( i + 1, ifac[i + 1] );
}
inline int comb ( const int n, const int m ) {
return n < m ? 0 : mul ( fac[n], mul ( ifac[m], ifac[n - m] ) );
}
inline void NTT ( const int n, int* A, const int flg ) {
rep ( i, 0, n - 1 ) if ( i < rev[i] ) iswp ( A[i], A[rev[i]] );
for ( int i = 2, stp = 1; i <= n; i <<= 1, stp <<= 1 ) {
int w = mpow ( G, ( MOD - 1 ) / i );
if ( !~flg ) w = inv ( w );
for ( int j = 0; j < n; j += i ) {
for ( int k = j, wk = 1; k < j + stp; ++k, wk = mul ( wk, w ) ) {
int ev = A[k], ov = mul ( wk, A[k + stp] );
A[k] = add ( ev, ov ), A[k + stp] = sub ( ev, ov );
}
}
}
if ( !~flg ) {
int in = inv ( n );
rep ( i, 0, n - 1 ) A[i] = mul ( in, A[i] );
}
}
int main () {
n = rint (), m = rint (), s = rint ();
init ( imax ( n, m ) );
rep ( i, 0, m ) w[i] = rint ();
int mx = imin ( m, n / s );
for ( int i = 0, fp = 1; i <= mx; ++i, fp = mul ( fp, fac[s] ) ) {
f[i] = mul ( fac[i], mul ( mul ( comb ( m, i ), mpow ( m - i, n - i * s ) ),
mul ( fac[n], inv ( mul ( fp, fac[n - i * s] ) ) ) ) );
}
rep ( i, 0, mx ) {
g[i] = ( ( mx - i ) & 1 ? sub : add )( 0, inv ( fac[mx - i] ) );
}
int len = 1, lgv = 0;
for ( ; len < mx + 1 << 1; len <<= 1, ++lgv );
rep ( i, 0, len - 1 ) rev[i] = ( rev[i >> 1] >> 1 ) | ( ( i & 1 ) << lgv >> 1 );
NTT ( len, f, 1 ), NTT ( len, g, 1 );
rep ( i, 0, len - 1 ) f[i] = mul ( f[i], g[i] );
NTT ( len, f, -1 );
int ans = 0;
rep ( i, 0, mx ) {
ans = add ( ans, mul ( w[i], mul ( f[i + mx], ifac[i] ) ) );
}
printf ( "%d\n", ans );
return 0;
}
Solution -「HAOI 2018」「洛谷 P4491」染色的更多相关文章
- [洛谷P4491] [HAOI2018]染色
洛谷题目链接:[HAOI2018]染色 题目背景 HAOI2018 Round2 第二题 题目描述 为了报答小 C 的苹果, 小 G 打算送给热爱美术的小 C 一块画布, 这块画布可 以抽象为一个长度 ...
- 「区间DP」「洛谷P1043」数字游戏
「洛谷P1043」数字游戏 日后再写 代码 /*#!/bin/sh dir=$GEDIT_CURRENT_DOCUMENT_DIR name=$GEDIT_CURRENT_DOCUMENT_NAME ...
- Solution -「JSOI 2019」「洛谷 P5334」节日庆典
\(\mathscr{Description}\) Link. 给定字符串 \(S\),求 \(S\) 的每个前缀的最小表示法起始下标(若有多个,取最小的). \(|S|\le3\time ...
- Solution -「洛谷 P4372」Out of Sorts P
\(\mathcal{Description}\) OurOJ & 洛谷 P4372(几乎一致) 设计一个排序算法,设现在对 \(\{a_n\}\) 中 \([l,r]\) 内的元素排 ...
- Solution -「POI 2010」「洛谷 P3511」MOS-Bridges
\(\mathcal{Description}\) Link.(洛谷上这翻译真的一言难尽呐. 给定一个 \(n\) 个点 \(m\) 条边的无向图,一条边 \((u,v,a,b)\) 表示从 ...
- Solution -「APIO 2016」「洛谷 P3643」划艇
\(\mathcal{Description}\) Link & 双倍经验. 给定 \(n\) 个区间 \([a_i,b_i)\)(注意原题是闭区间,这里只为方便后文描述),求 \(\ ...
- 「洛谷4197」「BZOJ3545」peak【线段树合并】
题目链接 [洛谷] [BZOJ]没有权限号嘤嘤嘤.题号:3545 题解 窝不会克鲁斯卡尔重构树怎么办??? 可以离线乱搞. 我们将所有的操作全都存下来. 为了解决小于等于\(x\)的操作,那么我们按照 ...
- 「洛谷3338」「ZJOI2014」力【FFT】
题目链接 [BZOJ] [洛谷] 题解 首先我们需要对这个式子进行化简,否则对着这么大一坨东西只能暴力... \[F_i=\sum_{j<i} \frac{q_iq_j}{(i-j)^2}-\s ...
- 「BZOJ2733」「洛谷3224」「HNOI2012」永无乡【线段树合并】
题目链接 [洛谷] 题解 很明显是要用线段树合并的. 对于当前的每一个连通块都建立一个权值线段树. 权值线段树处理操作中的\(k\)大的问题. 如果需要合并,那么就线段树暴力合并,时间复杂度是\(nl ...
随机推荐
- mysql数据库读写分离教程
注意:实现MySQL读写分离的前提是我们已经将MySQL主从复制配置完毕 一.Mycat实现读写分离安装和配置 架构规划: 192.168.201.150 master 主节点 192.168. ...
- Spark-local本地环境搭建
注:图片如果损坏,点击文章链接:https://www.toutiao.com/i6815385772254822919/ 承接上一个文档<Spark源码编译> 解压spark编译好的压缩 ...
- Word2010制作自动目录
原文链接:https://www.toutiao.com/i6488296610873737741/ 原文从网上复制: 查看"开始"选项卡,"样式"功能组,我们 ...
- Maven+ajax+SSM实现删除
转载自:https://www.cnblogs.com/kebibuluan/p/9020381.html 3.尚硅谷_SSM高级整合_使用ajax操作实现删除的功能 点击删除的时候,要删除联系人,这 ...
- Keil MDK STM32系列(四) 基于抽象外设库HAL的STM32F401开发
Keil MDK STM32系列 Keil MDK STM32系列(一) 基于标准外设库SPL的STM32F103开发 Keil MDK STM32系列(二) 基于标准外设库SPL的STM32F401 ...
- [javaweb]strut2-001漏洞分析
Strut2-001 漏洞描述 框架解析JSP页面标签时会对用户输入的Value值获取,在获取对应的Value值中递归解析%{.}造成了二次解析,最终触发表达式注入漏洞,执行任意代码 影响版本 2.0 ...
- C# 给PDF文档设置过期时间
我们可以给一些重要文档或者临时文件设置过期时间和过期信息提示来提醒读者或管理者文档的时效性,并及时对文档进行调整.更新等.下面,分享通过C#程序代码来给PDF文档设置过期时间的方法. 引入dll程序集 ...
- 【刷题-LeetCode】188 Best Time to Buy and Sell Stock IV
Best Time to Buy and Sell Stock IV Say you have an array for which the i-th element is the price of ...
- NIO-java.nio.ByteBuffer中flip、rewind、clear方法的区别
Java NIO中的Buffer用于和NIO通道进行交互.如你所知,数据是从通道读入缓冲区,从缓冲区写入到通道中的. 缓冲区本质上是一块可以写入数据,然后可以从中读取数据的内存.这块内存被包装成NIO ...
- pytest文档2-用例执行
用例设计原则 1.文件名以test_******.py文件和*******_test.py 2.以test_****开头的函数 3.以Test***开头的类 4.以test_*****开头的方法 5. ...