OI 容斥原理 学习笔记

1 基本形式

$$|\bigcup _{i=1}^n{A}_i|=\sum _{k=1}^n(-1{)}^{k-1}\sum _{1\le i_1<i_2<i_3<\cdots <i_k\le n}|A_{i_1}\cap A_{i_2}\cap A_{i_3}\cap \cdots \cap A_{i_k}|$$

1.1 证明

1. 数学归纳法
2. 考虑每一个元素 $x$ 的贡献

2 应用

2.1 错排

求把 $1-n$ 放在 $n$ 个位置上，使得 $i$ 位置上放的数不是 $i$ 的方案数。

2.1.1 递推做法

$d_n=(n-1)(d_{n-1}+d_{n-2})$

思路：考虑第 $n$ 个位置上的数放在 $x$ 处，分 $x$ 放在 $n$ 和放在 $y(y\ne n)$ 两种情况考虑。

2.1.2 容斥做法

令 $A_x$ 为 $i\in x|a_i=i$ 的排列集，

则

a n s = n ! − ∣ A 1 ∪ A 2 ∪ ⋯ ∪ A n ∣ = n ! − ∑ ∣ A { i } ∣ + ∑ ∣ A { i , j } ∣ − ∑ ∣ A { i , j , k } ∣ + … = n ! − ( n 1 ) ( n − 1 ) ! + ( n 2 ) ( n − 2 ) ! − ( n 3 ) ( n − 3 ) ! + … = ∑ i = 2 n ( n i ) ( − 1 ) i ( n − i ) ! \begin{aligned} ans &=n! - |A_1 \cup A_2 \cup \dots \cup A_n |\\ &= n! - \sum |A_{\{i\}}| + \sum |A_{\{i,j\}}| - \sum |A_{\{i,j,k\}}| + \dots \\ &= n! - \binom{n}{{}1} (n-1)! + \binom{n}{2}(n-2)! - \binom{n}{3}(n-3)! +\dots \\ &= \sum \limits _{i=2} ^n \binom{n}{i}(-1)^i(n-i)! \end{aligned} ans​=n!−∣A1​∪A2​∪⋯∪An​∣=n!−∑∣A{i}​∣+∑∣A{i,j}​∣−∑∣A{i,j,k}​∣+…=n!−(1n​)(n−1)!+(2n​)(n−2)!−(3n​)(n−3)!+…=i=2∑n​(in​)(−1)i(n−i)!​

2.2 DAG 计数

计算有 $n$ 个点的有向无环图（可以不连通）的个数。

2.2.1 做法

有向无环图中，必有至少一个点入度为 $0$。

令 $A_x$ 为 $i\in x|in(i)=0$ 的图的集合（$in(x)$ 表示 $x$ 的入度），$B_x$ 为 $\sum A_i(|i|=x)$，$f(n)$ 为题目所求，

则

$$\begin{array}{rl}f(n)& =|A_1\cup A_2\cup \cdots \cup A_n|\\ & =\sum _{i=1}^n(-1{)}^{i-1}{C}_n^i{B}_i\end{array}$$

考虑有 $i$ 个点入度为 $0$ 的情况，将整个图分为入度为 $0$ 的点和其他点两部分，那么只需要考虑这两部分相连的情况，所以

$$B_i=2^{i(n-i)}f(n-i)$$

代入上面的式子就可以得出

$$f(n)=\sum _{i=1}^n(-1{)}^{i-1}{C}_n^i{2}^{i(n-i)}f(n-i)$$

直接递推求解。

3 例题

3.1 CF585E

3.1.1 题意

给定一个整数序列 $A_{1,2,\dots ,n}$，求：在序列中选出一个集合包含若干个 $\gcd \ne 1$ 的数，再在剩下的数中选一个数 $A_y$ 使得 $A_y$ 与之前选出的集合中所有数互质的方案数。

$n\le 5\times 10^5,A_i\in [2,10^7]$。

3.1.2 解法

1 考虑只选一些数使得 $\text{gcd}\ne 1$。

令 $cnt[x]$ 表示序列 $A$ 中 $x$ 的倍数的个数，$B_x$ 表示选出的数集满足 $x|\gcd$ 的方案集合。

显然 $|B_x|=2^{cnt[x]}-1$（不能选空集，所有数取或不取）且 $B_{ij}\in B_i\cup B_j$，$B_i\cap B_j=B_{ij}$。

所以有

$$\begin{array}{rl}ans& =|B_2\cup B_3\cup B_4\cup \cdots \cup B_{inf}|\\ & =|B_2\cup B_3\cup B_5\cup B_7\cup \cdots \cup B_{P_{last}}|\\ & =\sum |B_{P_i}|-\sum |B_{P_i{P}_j}|+\sum |B_{P_i{P}_j{P}_k}|-\dots \\ & =\sum _{x=2}^{inf}\mu (x)|B_x|\\ & =\sum _{x=2}^{inf}\mu (x)(2^{cnt[x]}-1)\end{array}$$

（$\mu (x)$ 是莫比乌斯函数）

2 考虑先选出一个数，再选出符合条件的数的集合。

令 $G_{A_i}$ 表示在除 $A_i$ 以外的数中，选出的数集满足 $\gcd |A_i$ 的个数。

设此时的方案数为 $res$，则

$$\begin{gathered} res=\sum _{i=1}^n(ans-|G_{A_i}|) \\ |G_{A_i}|=\sum _{x|A_i}\mu (x)|B_x| \end{gathered}$$

所以

$$\begin{array}{rl}res& =\sum _{i=1}^{n}ans-\sum _{i=1}^n|G_{A_i}|\\ & =n\sum _{i=2}^{inf}\mu (x)|B_x|-\sum _{i=1}^n\sum _{x>1\&x|A_i}\mu (x)|B_x|\\ & =n\sum _{i=2}^{inf}\mu (x)|B_x|-\sum _{x=2}^{inf}\sum _{\forall i:x|A_i}\mu (x)|B_x|\\ & =\sum _{i=2}^{inf}\mu (x)|B_x|(n-cnt[x])\\ & =\sum _{i=2}^{inf}\mu (x)(2^{cnt[x]}-1)(n-cnt[x])\end{array}$$

直接递推求解即可，时间复杂度 $O(inf=10^7)$。

3.2 Loj 2541