NOI2020 D1T3 时代的眼泪 口胡题解

Description

给定一个长度为 $n$ 的排列 $p$，每次给定 $l_1,r_1,l_2,r_2$，求 $[l_1,r_1]$ 中存在多少对 $(i,j)$，满足：

① $i\le j$
② $p_i\le p_j$
③ $p_i\in [l_2,r_2]$，$p_j\in [l_2,r_2]$

$1\le n,q\le 10^5$，时限 $\text{5s}$，空限 $\text{1GB}$。

Solution

考虑分块，令块长为 $\lfloor \sqrt{n}\rfloor =B$。

对于一次形如 $L,R,l,r$ 的询问，我们将 $[L,R]$ 划分为 $[L_1,p-1]$，$[p,q]$ 与 $[q+1,R]$ 三个部分，分别表示开头处的零散块，中间的整块与末尾处的零散块。

现在，关键在于快速处理：

• $[p,q]$ 对 $[p,q]$ 的贡献
  • 跨块的贡献
  • 块内的贡献
• $[L,p-1]$ 对 $[q+1,R]$ 的贡献
• $[L,p-1]$ 对 $[p,q]$ 的贡献
• $[p,q]$ 对 $[q+1,R]$ 的贡献
• $[L,p-1]$ 对 $[L,p-1]$ 的贡献
• $[q+1,R]$ 对 $[q+1,R]$ 的贡献。

Part 1

先考虑 $[p,q]$ 对 $[p,q]$ 的贡献中跨块的贡献。

令 $f_{i,j}$ 表示，第 $i$ 个块中 $j$ 出现的次数。
令 $g_{i,j}$ 表示，前 $i$ 个块中 $j$ 出现的次数。

不难得到这一部分的贡献是

$$\sum _{x=c}^d\sum _{i=l}^r{f}_{x,i}\times ((g_{d,r}-g_{d,i})-(g_{x,r}-g_{x,i}))$$

这里 $c,d$ 表示 $p$ 与 $q$ 所在的块的编号。

考虑展开成四个部分，则原式可以写成下面的四个式子之和。现在关键在于计算下面的四个式子。

Part 1.1

$$\sum _{x=c}^d\sum _{i=l}^r{f}_{x,i}\times g_{d,r}$$

显然，随着 $x$ 与 $i$ 的变化 $g_{d,r}$ 不变，于是我们可以将 $g_{d,r}$ 提到外面。

我们先预处理，对每一个 $f_x$ 做前缀和；查询的时候，枚举 $x$，$O(1)$ 求出 ${\sum }_{x=c}^d{f}_{x,i}$ 并累加即可。

Part 1.2

$$-\sum _{x=c}^d\sum _{i=l}^r{f}_{x,i}\times g_{d,i}$$

交换一下 $\sum$

$$-\sum _{i=l}^r{g}_{d,i}\sum _{x=c}^d{f}_{x,i}$$

对于每一个 $i$，预处理出 $f_{x,i}$ 的前缀和，查询的时候枚举 $i$，$O(1)$ 求出里面的 $\sum$，乘上 $g_{d,i}$ 并累加即可。

Part 1.3

$$-\sum _{x=c}^d\sum _{i=l}^r{f}_{x,i}\times g_{x,r}$$

同理。

Part 1.4

$$\sum _{x=c}^d\sum _{i=l}^r{f}_{x,i}\times g_{x,i}$$

维护 $p_{x,i}=f_{x,i}\times g_{x,i}$ 的前缀和即可。

---

从而，Part 1 可以单次 $O(\sqrt{n})$ 算出。

Part 2

考虑 $[p,q]$ 对 $[p,q]$ 的贡献中块内部产生的贡献。显然，这一部分中块与块是独立的，这启发我们去预处理。

我们可以对于每一个块，将其内部的 $a$ 做内部离散化并记录两个数组 $lis1,lis2$。对于一个真实值 $i$，该块内不超过它的最大数即为 $lis{1}_i$，该块内不小于它的最小数即为 $lis{2}_i$。

离散化后，我们在块内找到每一对正序对，并将它们存储在一个 $B\times B$ 的二维表里面，并将这个表做二维前缀和；注意，表的第 $i$ 行第 $j$ 列表示满足 $a_x=i,a_y=j$ 的逆序对 $(a_x,a_y)$ 的数量。查询的时候，枚举块，令 $l^{\prime }=lis{1}_{l2}$，$r^{\prime }=li{s}_{r2}$，二维矩阵 $[l^{\prime },r^{\prime }][l^{\prime },r^{\prime }]$ 的和即为该块的贡献。

从而，Part 2 单次 $O(\sqrt{n})$ 求出。

Part 3

我们在 $[l_1,p-1]$ 中枚举一个 $a_i(a_i\in [l_2,r_2])$，然后在 $[q+1,r_1]$ 所在块中查表得到 $l^{\prime }=lis{1}_{a_i+1},r^{\prime }=lis{2}_{r_2}$。

现在，我们需要求出，$[q+1,r_1]$ 中存在多少个在 $[l^{\prime },r^{\prime }]$ 中的数（注意，已经离散化）。我们对于每个块，对于每个 $i$，预处理一个长度为 $B$ 的数组 $li{s}_3$ 表示 $i$ 在该块各个位置的出现情况 $0/1$，并将其做前缀和，从而每次可以 $O(1)$ 查询。

从而，Part 3 可以在单次 $O(\sqrt{n})$ 的复杂度内求出。

Part 4

枚举一个 $[l_1+1,p-1]$ 中的数 $a_i(a_i\in [l_2,r_2])$，然后调用 $g$ 的前缀和数组即可。

从而，Part 4 可以在单次 $O(\sqrt{n})$ 的复杂度内求出。

Part 5

与 Part 4 同理，单次 $O(\sqrt{n})$。

Part 6

枚举一个 $[l_1+1,p-1]$ 中的位置 $i$，然后在 $li{s}_3$ 的前缀和数组里面做查询即可。

单次 $O(\sqrt{n})$。

Part 7

与 Part 6 同理。

值得注意的是，本题也需要特判“恰好在一个块内的查询”的情况，且此情况可以采用与 Part 6/7 类似的方法进行单次 $O(\sqrt{n})$ 的计算。

---

综上所述，时间复杂度 $O((n+q)\sqrt{n})$，空间复杂度 $O(n\sqrt{n})$，常数非常大。

Summary

我们开了许多大数组，这里有必要对空间进行规划。

• $f$，空间为 $\frac{n^2}{B}$
• $g$，空间为 $\frac{n^2}{B}$
• $f_{i,j}$，固定 $i$ 时的前缀和，空间为 $\frac{n^2}{B}$
• $f_{i,j}$，固定 $j$ 时的前缀和，空间为 $\frac{n^2}{B}$
• $p_{i,j}$，固定 $i$ 时的前缀和，空间为 $\frac{n^2}{B}$
• $lis1$，空间为 $\frac{n^2}{B}$
• $lis2$，空间为 $\frac{n^2}{B}$
• 每个块内都有一个二维矩阵的前缀和数组，空间为 $nB$
• $lis3$ 的前缀和，空间为 $nB$

当 $7\times \frac{n^2}{B}$ 与 $2nB$ 十分接近时，空间最优。解方程，不难得到

$$B=\frac{\sqrt{14n}}{2}$$

带入 $n=10^5$，空间总和是 $236643311$。由于空间限制是 $\text{1GB}$，所以可以通过。