ABC366D 题解

推荐在 cnblogs 上阅读

Solution

题意简述

给你一个正整数 $N$ 和 $N^3$ 个非负整数，表示为 $A_{x,y,z}$ 其中 $1\le x,y,z\le N$ 。

您将得到以下格式的 $Q$ 个查询，必须按顺序处理。

对于第 $i$ 次查询 $(1\le i\le Q)$ ，您将得到一个整数元组 $(L{x}_i,R{x}_i,L{y}_i,R{y}_i,L{z}_i,R{z}_i)$ ，其中 $1\le L{x}_i\le R{x}_i\le N$ , $1\le L{y}_i\le R{y}_i\le N$ , 和 $1\le L{z}_i\le R{z}_i\le N$ 。求

$$\sum _{x=L{x}_i}^{R{x}_i}\sum _{y=L{y}_i}^{R{y}_i}\sum _{z=L{z}_i}^{R{z}_i}{A}_{x,y,z}$$

题解

简单三维前缀和。考虑一下容斥关系就好了。

code

容斥知识补充

既然题解写都写了，写详细一点。

首先引入容斥的最经典的就是这张韦恩图（来自 OI-wiki）：

现在我想知道 $|A\cup B\cup C|$，并不是简单粗暴的三个集合大小相加，因为互相有重叠部分。正确答案是：$|A\cup B\cup C|=|A|+|B|+|C|-|A\cap B|-|A\cap C|-|B\cap C|+|A\cap B\cap C|$。相信还是很好理解的这里不赘述了。

把它推广到一般情况，就是我们熟悉的容斥原理了。

定义

$$\begin{gathered} \left|\bigcup _{i=1}^n{S}_i\right|=\sum _i\left|S_i\right|-\sum _{i<j}\left|S_i\cap S_j\right|+\sum _{i<j<k}\left|S_i\cap S_j\cap S_k\right|-\dots \\ +(-1{)}^{m-1}\sum _{a_i<a_{i+1}}\left|\bigcap _{i=1}^m{S}_{a_i}\right|+\cdots +(-1{)}^{n-1}\left|S_1\cap \cdots \cap S_n\right| \end{gathered}$$

整理，得

$$\left|\bigcup _{i=1}^n{S}_i\right|=\sum _{m=1}^n(-1{)}^{m-1}\sum _{a_i<a_{i+1}}\left|\bigcap _{i=1}^m{S}_{a_i}\right|$$

证明

我们计算每个元素出现的次数，对于 $x$，假设它出现在 $T_1,\dots ,T_m$ 的集合中。

C n t = ∣ { T i } ∣ + ⋯ + ( − 1 ) k − 1 ∣ { ⋂ i = 1 k T a i ∣ a i < a i + 1 } ∣ + ⋯ + ( − 1 ) m − 1 ∣ { T 1 ∩ ⋯ ∩ T m } ∣ = ( m 1 ) − ( m 2 ) + ⋯ + ( − 1 ) m − 1 ( m m ) = ( m 0 ) − ∑ i = 0 m ( − 1 ) i ( m i ) = 1 − ( 1 − 1 ) m = 1 \begin{aligned} Cnt&=|\{T_i\}|+\dots+(-1)^{k-1}\left|\left\{\bigcap_{i=1}^kT_{a_i}|a_i<a_{i+1}\right\}\right|+\dots+(-1)^{m-1}|\{T_1\cap\dots\cap T_m\}|\\ &=\binom{m}{1}-\binom{m}{2}+\dots+(-1)^{m-1}\binom{m}{m}\\ &=\binom{m}{0}-\sum\limits_{i=0}^m(-1)^i\binom{m}{i}\\ &=1-(1-1)^m=1 \end{aligned} Cnt​=∣{Ti​}∣+⋯+(−1)k−1 ​{i=1⋂k​Tai​​∣ai​<ai+1​} ​+⋯+(−1)m−1∣{T1​∩⋯∩Tm​}∣=(1m​)−(2m​)+⋯+(−1)m−1(mm​)=(0m​)−i=0∑m​(−1)i(im​)=1−(1−1)m=1​

每个元素出现次数为 $1$，合并起来就是并集了，证毕。