容斥原理qwq

_nebula_

已于 2023-10-11 21:46:20 修改

阅读量78

点赞数

文章标签：算法

于 2023-09-23 18:36:25 首次发布

本文链接：https://blog.csdn.net/FuDuoDuo_qwq/article/details/133213570

版权

update(2023.9.26) 增加例题 ABC321G
update(2023.10.10) 增加例题 ARC160D

感觉这篇博客讲的很清楚，下图截图自前面的博客，所以我不在赘述了。

另外，计数题的要求就是不重不漏，对于不重复，直接减去之前已经算好的只包含当前状态的局部的状态的方案数，也就是容斥的本质。
所以很多包含 dp 的容斥题的 dp 状态就是全集的一个子集。最终状态就是全集。转移就是“枚举子集”。

CF449D Jzzhu and Numbers

题目

直接对题目条件容斥。

CF1559E Mocha and Stars

题目

在 gcd 方面，容斥的 trick 就是设 $d p [i]$ 表示 gcd 等于 $i$ 的倍数的方案数， $an s [i]$ 表示 gcd 等于 i 的方案数。求出 $d p [i]$ ， $ans[i]=dp[i]-\sum_{i|j} ans[j]$ 。

对于这道题，考虑如何求出 $d p [i]$ 。首先枚举 gcd，假设 gcd 等于 $g$ ，然后设 $f [i] [j]$ 表示前 $i$ 个数，数字和等于 $j\times g$ 的方案数。转移时间复杂度是 $O(n\space ln\space m)$ ，总时间复杂度等于 $O(nm\space ln\space m)$ 。

CF1591F Non-equal Neighbours

题目

首先看到不等想到容斥，但是直接容斥肯定解决不了 $n\le 2\times 10^5$ ~~（至少我不会）~~ 。所以考虑设一个很容斥的 dp： $f [i]$ 表示前 $i$ 个数满足条件的方案数。那么 $f[i]\times min_{j=i+1}^n a_j$ 就表示前 $i - 1$ 个数满足 $b_i\neq b_{i+1}$ ，后面的数满足 $b_i=b_{i+1}$ 的方案数。

于是转移就是 $f[i]=\sum_{j=0}^{i-1} (-1)^{i-j+1} f[j]\times min_{k=j+1}^i a_k$ 。

考虑优化，令 $p os [i]$ 表示 $[1, i)$ 最后一个小于 $a_i$ 的位置。

$f[i]=\sum_{j=0}^{i-1} (-1)^{i-j+1}f[j]\times min_{k=j+1}^i a_i$

$f[i]=(a_i\sum_{j=pos[i]+1}^{i-1} (-1)^{i-j+1} f[j])+(-1)^{i-pos[i]}\sum_{j=0}^{pos[i]} (-1)^{pos[i]-j+1} f[j]\times min_{k=j+1}^i a_k$

$\therefore f[i]=(a_i\sum_{j=pos[i]+1}^{i-1} (-1)^{i-j+1} f[j])+(-1)^{i-pos[i]} f[pos[i]]$

记一个正负交替的前缀和即可。

luogu P6846 [CEOI2019] Amusement Park

题目
设 $f [s]$ 包含表示点集的 DAG 数。转移就是枚举当前 DAG 中度为 $0$ 的点集 $t$ ，然后 $f[s]\leftarrow f[s\oplus t]$ 。
但是当前加入的度为 $0$ 的集合不一定都会向上一次加入的度为 $0$ 集合连边，所以会算重。所以转移就是 $f[s]=\sum_{t\subseteq s} (-1)^{|t|+1}f[s\oplus t]$ 。时间复杂图 $O(3^n)$ 。

ABC321G Electric Circuit

题目
场上读题都读了半天，好菜。
首先设 $h_s$ 表示点集 $s$ 连通的方案数。需要注意的是，点集 $s$ 要满足 $\sum_{i=1}^m[R_i\in s]=\sum_{i=1}^m[B_i\in s]$ 。满足上面条件后，假设 $cnt_s=\sum_{i=1}^m[R_i\in s]$ ，然后我们有转移 $h_s=cnt_s!-\sum_{t\in s} h_t\times cnt_{s\oplus t}!$ 。
但是可以发现这样转移会出现重复，那么我们钦定 $lowbit(s)\in t$ 也就是点集 $s$ 中最小的元素被包含在转移的最后一个连通块里。
然后设 $f_s$ 表示对于点集 $s$ 的虽有连法的连通数总和， $g_s$ 表示点集 $s$ 的方案数。
$f_s=\sum_{t\in s} f_{s\oplus t}\times h_t+g_{s\oplus t}\times h_t$ 。
$g_s=\sum_{t\in s} g_{s\oplus t}\times h_t$ 。
上面的转移也会出现重复，仍然用前面的方法解决。

ARC160D Mahjong

题目

我们考虑不去对最终的序列求方案数，而是统计可行的操作序列个数，但是这样会算重。那么我们考虑钦定包含 $i$ 的区间加 $1$ 的操作只能执行 $< k$ 次，那么就没有问题了。
所以现在问题转换成对一个长度为 $2 n - k + 1$ 的序列统计个数，要求对于 $i\in[1,2n-k+1]$ ，满足 $a_i\ge0$ ， $\sum_{i=1}^{2n-k+1}a_i=\frac{m}{k}$ ，且 $i\in[1,n-k+1]$ ， $a_i<k$ 。
对满足限制的个数容斥，然后是插板法，式子大概长成这样：
$ans=\sum_{i=0}^{n-k+1}(-1)^i\binom{n-k+1}{i}\binom{\frac{m}{k}-i\times k+2n-k}{2n-k}$