历年各赛事真题选做(一)

历年各赛事真题选做(一)

\(\newcommand \oper \operatorname\)\(\newcommand \seq \mathcal\) \(\newcommand \m \mathbf\)\(\newcommand \txt \texttt\)

\(\text{By DaiRuiChen007}\)

推荐阅读:

  • A. [九省联考 2018] 秘密袭击
  • E. [HNOI/AHOI2018] 寻宝游戏
  • K. [APIO2020] 有趣的旅途
  • Q. [WC2018] 州区划分
  • V. [HNOI/AHOI2018] 排列
  • Z. [省选联考 2021 A 卷] 矩阵游戏

A. [九省联考 2018] 秘密袭击

Problem Link

I. 朴素的 dp

首先,我们知道答案就是求所有联通块中的第 \(k\) 大值和

第一步转化,我们可以拆分贡献,对于每个 \(i\) 求出:连通块中第 \(k\) 大数为 \(i\) 的方案数,把方案数乘以 \(i\) 再求和即可

第二步转化,我们可以把乘以 \(i\) 的贡献拆成 \(i\) 次计算的贡献,即:对所有 \(i\),统计连通块中的 \(k\) 大数 \(\ge i\) 的方案数,这样每个 \(i\) 恰好被计算 \(i\)

第三步转化,我们可以把“第 \(k\) 大数 \(\ge i\)”转化成“值 \(\ge i\) 的数 \(\ge k\) 个”的条件,即:对于所有 \(i\),求出 \(\ge i\) 的数在连通块中出现次数 \(\ge k\) 的方案数

在转化之后,我们可以设计一个显然的 dp 来解决这个问题,用 \(f_{u,i,j}\) 表示以 \(u\) 为根的子树中值 \(\ge i\) 的数出现恰好 \(j\) 次的连通块数量

为了防止算重,我们可以强制要求 \(f_{u,i,j}\) 中统计的连通块一定包含 \(u\),从而防止重复计算

\(f_{u,i,j}\) 的转移比较显然,直接从对于每个 \(u\) 的儿子 \(v\),用 \(f_{v,i,j'}\) 以一个卷积的形式往上转移

为了更加方便统计答案,我们不妨再定义一个 \(g_{u,i,j}\) 表示所有 \(u\) 的子树中的 \(v\)\(f_{v,i,j}\) 的和,即非强制选定 \(u\) 时的方案数,\(g\) 的转移比较简单,直接从 \(u\) 的所有儿子处转移 \(g\) 值并且转移 \(u\) 本身的 \(f\)

那么,答案就是 \(\sum_{i=1}^n\sum_{j=k}^n g_{\oper{root},i,j}\),即对于每个 \(i\in[1,n]\),统计选择值 \(\ge i\) 的节点,选择了 \(k\sim n\) 次的方案数再累加求和

II. 多项式技巧维护 dp

注意到 \(f\) 的转移很像对 \(j\) 一维卷积的形式,因此我们可以在 \(f_{u,i},g_{u,i}\) 中用生成函数维护所有 \(f_{u,i,j},g_{u,i,j}\) 的值

不妨假设 \(F_{u,i}=\sum_{j=0}^n f_{u,i,j}\times x^j,G_{u,i}=\sum_{j=0}^n g_{u,i,j}\times x^j\)

那么我们就把状态转移方程转化为如下的形式:

\[\begin{aligned} F_{u,i}&=x^{[d_u\ge i]}\prod_{v\in \oper{son}(u)} (F_{v,i}+1)\\ G_{u,i}&=F_{u,i}+\sum_{v\in\oper{son}(u)} G_{v,i} \end{aligned} \]

那么在这一步之后,我们把答案转化成了 \(\sum_{i=1}^n G_{\oper{root},i}\)\(k\sim n\) 次项系数的和

不妨假设 \(H(x)=\sum_{i=1}^n G_{\oper{root},i}\),那么原问题只需要我们求出 \(H(x)\) 即可解决

注意到在树上多次进行多项式乘法的复杂度较高,难以接受,因此我们考虑用用拉格朗日插值法把多项式乘法转化成普通的乘法

即,对于每个 \(x_0=1,2,3,\dots,n+1\),我们把 \(x=x_0\) 带入所有 \(F_{u,i},G_{u,i}\) 的表达式,求出所有 \(F_{u,i},G_{u,i}\)\(x=x_0\) 处的点值,从而得到 \(H(x)\)\(x=x_0\) 处的点值

最终,我们通过 \(H(1)\sim H(n+1)\) 的点值运用拉格朗日插值法得到 \(H(x)\) 的系数表达,再把 \(k\sim n\) 次项的系数统计到答案中

因此,我们只需要对于每个 \(x_0\),维护如下的转移:

\[\begin{aligned} F_{u,i}&=x_0^{[d_u\ge i]}\prod_{v\in \oper{son}(u)} (F_{v,i}+1)\\ G_{u,i}&=F_{u,i}+\sum_{v\in\oper{son}(u)} G_{v,i} \end{aligned} \]

注意,此处转移方程式中的 \(F_{u,i},G_{u,i}\) 都是数值而非多项式

III. 线段树加速转移

注意到在状态转移方程中,每个 \(i\) 对答案的实际影响很小,因此可以对于每个 \(u\),把所有的 \(F_{u,i},G_{u,i}\) 看成两个长度为 \(n\) 的序列 \(\seq F_u,\seq G_u\) 总体维护

那么整个状态转移的过程可以拆解成如下的步骤:

  • \(\seq F_u[1\sim n]\gets 1\)
  • 对于所有 \(i\in[1,a_u]\),令 \(\seq F_u[i]\gets \seq F_u[i]\times x_0\)
  • 枚举 \(u\) 的所有儿子 \(v\)
    • 对于所有 \(i\in [1,n]\),令 \(\seq F_v[i]\gets \seq F_v[i]+1\)
    • 对于所有 \(i\in [1,n]\),令 \(\seq F_u[i]\gets \seq F_u[i]\times \seq F_v[i],\seq G_u[i]\gets \seq G_u[i]+\seq G_v[i]\)
  • 对于所有 \(i\in[1,n]\),令 \(\seq G_u[i]\gets \seq G_u[i]+\seq F_u[i]\)

具体的维护过程可以考虑用对于每一个节点 \(i\) 维护两个关键字 \((f,g)\)

而我们发现在序列上每一次的操作都可以表示成形如 \((f,g)\gets (a\times f+b,c\times f+g+d)\) 的形式

我们可以从矩阵乘法的角度来理解这个标记的变换,容易构造出如下的转移矩阵:

\[\begin{bmatrix} a&c&0\\ 0&1&0\\ b&d&1\\ \end{bmatrix} \times \begin{bmatrix} 0&0&0\\ 0&1&0\\ f&g&1 \end{bmatrix} = \begin{bmatrix} 0&0&0\\ 0&1&0\\ a\times f+b&c\times f+g+d&1 \end{bmatrix} \]

更一般地,而我们可以把任意两个转移矩阵的合并看成如下的过程:

\[\begin{bmatrix} a_2&c_2&0\\ 0&1&0\\ b_2&d_2&1\\ \end{bmatrix} \times \begin{bmatrix} a_1&c_1&0\\ 0&1&0\\ b_1&d_1&1\\ \end{bmatrix} = \begin{bmatrix} a_1\times a_2&a_1\times c_2+c_1&0\\ 0&1&0\\ a_2\times b_1+b_2&c_2\times b_1+d_1+d_2&1 \end{bmatrix} \]

注意到上面的过程等价于在原有的 \((a_1,b_1,c_1,d_1)\) 的标记上复合一个 \((a_2,b_2,c_2,d_2)\),由于矩阵左乘的过程具有结合律,因此我们可以用线段树来维护这些 \((a,b,c,d)\) 的标记,并且 \(f=b,g=d\)

因此我们可以得到一种新的运算 \(\oplus\)

\[(a_1,b_1,c_1,d_1)\otimes(a_2,b_2,c_2,d_2)=(a_1\times a_2,b_1\times a_2+b_2,a_1\times c_2+c_1,b_1\times c_2+d_1+d_2) \]

特别地,当我们只关心 \(f,g\) 的信息时,也可以写成:

\[(0,f,0,g)\otimes(a,b,c,d)=(0,a\times f+b,0,c\times f+g+d) \]

\(\oplus\) 的单位元为 \((1,0,0,0)\),任何标记 \(\otimes\) 上这个标记依然不改变自身的值

那么我们维护 \(\seq F_u,\seq G_u\) 的每一步的过程都可以用 \(\otimes\) 的形式表示出来:

  • \(\seq F_u[1\sim n]\gets 1\),等价于 \(\oplus (0,1,0,0)\)
  • 对于所有 \(i\in[1,a_u]\),令 \(\seq F_u[i]\gets \seq F_u[i]\times x_0\),等价于 \(\otimes(x_0,0,0,0)\)
  • 枚举 \(u\) 的所有儿子 \(v\)
    • 对于所有 \(i\in [1,n]\),令 \(\seq F_v[i]\gets \seq F_v[i]+1\),等价于 \(\otimes(1,1,0,0)\)
    • 对于所有 \(i\in [1,n]\),令 \(\seq F_u[i]\gets \seq F_u[i]\times \seq F_v[i],\seq G_u[i]\gets \seq G_u[i]+\seq G_v[i]\),等价于 \(\otimes(\seq F_v[i],0,0,\seq G_v[i])\)
  • 对于所有 \(i\in[1,n]\),令 \(\seq G_u[i]\gets \seq G_u[i]+\seq F_u[i]\),等价于 \(\otimes(1,0,1,0)\)

因此维护 \(\seq F_u,\seq G_u\) 的所有操作都可以写成区间 \(\otimes\) 某个标记或把两个序列对应位标记 \(\oplus\) 起来的形式

可以考虑使用线段树来优化,将这两种操作分别转化成线段树上的区间修改和合并两棵动态开点线段树,这样单次状态转移的时间复杂度就被优化到了可以接受的程度

IV. 最终处理与总结

最后,我们要在 \(\oper{root}\) 对应的线段树上查询每个节点上的标记的 \(d\) 值之和,作为对应的 \(H(x_0)\) 的值

接下来,我们只需要通过 \(H(1),H(2),\dots,H(n+1)\) 的点值还原一个 \(n\) 次多项式 \(H(x)\)

首先写出拉格朗日插值法的标准形式:

\[H(x)=\sum_{i=1}^{n+1} H(i)\prod_{j\ne i} \dfrac{x-j}{i-j} \]

然后把能够预处理的常数项提取出来,并且把可以公共计算的部分保留,从而对式子化简可得:

\[\begin{aligned} H(x) &=\sum_{i=1}^{n+1} H(i)\prod_{j\ne i} \dfrac{x-j}{i-j}\\[2ex] &=\sum_{i=1}^{n+1} \dfrac{H(i)}{\prod\limits_{j\ne i} (i-j)}\times \dfrac{\prod\limits_{j=1}^{n+1} (x-j)}{x-i} \end{aligned} \]

注意到:

  • 系数 \(\dfrac{H(i)}{\prod\limits_{j\ne i} (i-j)}\) 可以对每个 \(i\)\(\Theta(n)\) 的时间求出

  • 初始多项式 \(\prod\limits_{j=1}^{n+1} (x-j)\) 直接暴力卷积可以在 \(\Theta(n^2)\) 的时间内求出

  • 每次除以 \(x-i\) 的过程稍作模拟也可以用 \(\Theta(n)\) 执行单次修改

因此我们在 \(\Theta(n^2)\) 的时间复杂度内就可以插出 \(H(x)\) 的系数表达

最终,我们的总时间复杂度是 \(\Theta(n^2\log w)\),瓶颈在枚举 \(x_0\) 并且对于每个 \(x_0\) 在树上用线段树合并维护点值

Submission Link


B. [IOI2022] 数字电路

Problem Link

首先,我们把计数问题转成概率问题,假设每个点的阈值是对应范围内的一个随机整数,用 \(f_u\) 表示此时 \(u\) 的值为 \(1\) 的概率

考虑 \(u\) 的所有儿子 \(v\in\oper{son}(u)\),我们能求出 \(\oper{son}(u)\)\(1\) 的期望数量,设 \(c_u\) 表示 \(\oper{son}(u)\)\(1\) 的个数,那么:\(E(c_u)=\sum_{v\in\oper{son}(v)}f_v\)

接下来考虑如何通过 \(E(c_u)\) 推出 \(f_u\),我们可以把 \(E(c_u)\) 按定义对于不同的 \(c_u\) 给拆开,即令 \(g_{u,i}\) 表示 \(\oper{son}(u)\) 中出现恰好 \(i\)\(1\) 的概率,设 \(|\oper{son}(u)|=s_u\),那么我们有 \(\sum_{i=1}^{s_u} i\times g_{u,i}=E(c_u)\),这是根据期望的定义推出的

显然,为了求出 \(f_u\),我们只需对于每个 \(i\in[1,s_u]\) 求出:当 \(\oper{son}(u)\) 中有恰好 \(i\)\(1\) 时,\(u\) 的值为 \(1\) 的概率,显然,此时阈值应该 \(\le i\),而阈值的总取值共有 \(s_u\) 种,因此此时有 \(\dfrac i{s_u}\) 的概率使得 \(u\) 的值为 \(1\),将对应的 \(i\) 的贡献相加得到:\(f_u=\sum_{i=1}^{s_u}\dfrac i{s_u}\times g_{u,i}\)

联立两式,我们知道 \(f_u=\dfrac1{s_u}\sum_{i=1}^{s_u} i\times g_{u,i}=\dfrac{E(c_u)}{s_u}=\dfrac 1{s_u}\sum_{v\in\oper{son}(v)}f_v\),因此我们得到了 \(f_u\) 的转移

注意到 \(f_u\) 的转移形式是线性的,因此每个叶子结点对 \(f_0\) 的贡献也是线性的,准确来说:某个叶子 \(v\)\(f_0\) 的贡献是 \(\prod_{u\in\oper{path}(v,0)} \dfrac{1}{s_u}\)(忽略 \(s_u=0\) 的项,即不考虑叶子结点的贡献)

回到原来的计数问题上,某个叶子 \(v\) 对答案的贡献就是 \(\prod_{u\not\in\oper{path}(v,0)} s_u\),直接预处理出每个叶子对答案的贡献,注意到只有值为 \(1\) 的叶子对答案有贡献,因此用线段树维护区间反转,统计(有贡献的叶子的)整体和的操作即可

注意题目给出的模数不支持求逆元,因此可能要预处理子树内的 \(s_u\) 积,再用 dfs 逐个转移出每个叶子对答案的贡献

时间复杂度 \(\Theta((m+q)\log m)\)

Submission Link


C. [BJOI2018]二进制

Problem Link

首先我们对原有的条件进行分析,注意到 \(2^{2k}\bmod 3=1,2^{2k-1}\bmod 3=2\)

因此,对于一个二进制数 \(S\),我们用 \(x_0\) 表示 \(S_0+S_2+S_4+\cdots\),用 \(x_1\) 表示 \(S_1+S_3+S_5+\cdots\),那么 \(S\equiv x_0-x_1\pmod 3\)

又因为对于一个给定的区间,无论怎么重排,\(x_0+x_1\) 总是定值,那么我们可以通过对 \(x_0+x_1\) 的分析入手:

  • \(x_0+x_1\bmod 2=0\),直接使得 \(x_0=x_1\) 即可,即重排成一个形如 \(000\cdots001111\cdots1\) 的串其中串末尾恰好有偶数个 \(1\),而相邻的两个 \(1\) 可以看成 \(2^{k+1}+2^{k}=3\times 2^k\) 的形式,因此这个数必然是 \(3\) 的倍数
  • \(x_0+x_1\bmod 2=1\),由于 \(x_0-x_1\equiv x_0+x_1\equiv 1\pmod 2\)\(x_0-x_1\equiv S\equiv 3\pmod 3\),因此 \(|x_0-x_1|\) 至少是 \(3\),因此首先要保证 \(x_0+x_1\ge 3\),注意到我们可以构造一个形如 \(000\cdots 001111\cdots 11110101\) 的串,此时 \(x_0\)\(x_1\)\(3\) 个,且使用的 \(0\) 至少 \(2\) 个,可以证明不存在仅使用一个 \(0\) 而使得 \(x_0+x_1\bmod 3=0\) 的方案

我们发现对符合条件的区间计数比较麻烦,考虑反面考虑,对满足如下两个条件至少之一的串 \(S\) 进行计数:

显然,满足这两个条件中的任意一个的都是不合法的 \(S\),但由于这两个条件有相交部分,因此我们可以想到用容斥原理来计算,分别计算满足如下条件的 \(S\) 的数量:

  • \(S\) 中有奇数个 \(1\),且 \(0\) 的个数 \(\le 1\)
  • \(S\) 中只有一个 \(1\)
  • \(S\) 中只有一个 \(1\),且 \(0\) 的个数 \(\le 1\)

算出满足条件一、二的 \(S\) 的数量再减去满足条件三的 \(S\) 的数量即可

首先解决满足条件一的 \(S\) 的数量:

先考虑一个简单的 dp,用 \(f_{i,j,k}\) 表示以 \(i\) 结尾,其中 \(1\) 的个数 \(\bmod 2\) 的余数为 \(j\)\(0\) 的个数恰好为 \(k\)\(S\) 的数量

注意到只有 \(0\le k\le 1\) 的情况有用,而我们要统计的答案为 \(F_{i}=\sum_{j\le i} f_{i,1,0}+f_{i,1,1}\) 因此我们可以写出如下的转移方程:

  • \(a_i=0\) 时:

\[\begin{cases} f_{i,0,0}&=0\\ f_{i,0,1}&=f_{i-1,0,0}+1\\ f_{i,1,0}&=0\\ f_{i,1,1}&=f_{i-1,1,0}\\ F_{i}&=F_{i-1}+f_{i,1,0}+f_{i,1,1}=F_{i-1}+f_{i-1,1,0} \end{cases} \]

  • \(a_i=1\) 时:

\[\begin{cases} f_{i,0,0}&=f_{i-1,1,0}\\ f_{i,0,1}&=f_{i-1,1,1}\\ f_{i,1,0}&=f_{i-1,0,0}+1\\ f_{i,1,1}&=f_{i-1,0,1}\\ F_i&=F_{i-1}+f_{i,1,0}+f_{i,1,1}=F_{i-1}+f_{i-1,0,0}+f_{i-1,0,1}+1 \end{cases} \]

显然想到可以用矩阵来维护这个转移过程:

  • \(a_i=0\) 时:

\[\begin{bmatrix} f_{i-1,0,0}\\ f_{i-1,0,1}\\ f_{i-1,1,0}\\ f_{i-1,1,1}\\ F_{i-1}\\ 1 \end{bmatrix} \times \begin{bmatrix} 0&0&1&0&1&0\\ 0&0&0&1&1&0\\ 1&0&0&0&0&0\\ 0&1&0&0&0&0\\ 0&0&0&0&1&0\\ 0&0&1&0&1&1 \end{bmatrix} = \begin{bmatrix} 0\\ f_{i-1,0,0}+1\\ 0\\ f_{i-1,1,0}\\ F_{i-1}+f_{i-1,1,0}\\ 1 \end{bmatrix} \]

  • \(a_i=1\)

\[\begin{bmatrix} f_{i-1,0,0}\\ f_{i-1,0,1}\\ f_{i-1,1,0}\\ f_{i-1,1,1}\\ F_{i-1}\\ 1 \end{bmatrix} \times \begin{bmatrix} 0&0&1&0&1&0\\ 0&0&0&1&1&0\\ 1&0&0&0&0&0\\ 0&1&0&0&0&0\\ 0&0&0&0&1&0\\ 0&0&1&0&1&1 \end{bmatrix} = \begin{bmatrix} f_{i-1,1,0}\\ f_{i-1,1,1}\\ f_{i-1,0,0}+1\\ f_{i-1,0,1}\\ F_{i-1}+f_{i-1,0,0}+f_{i-1,0,1}+1\\1 \end{bmatrix} \]

初始向量为 \(\begin{bmatrix}0&0&0&0&0&1\end{bmatrix}\)

同理,对于第二个条件,我们也可以做类似的分析:

\(g_{i,j}\) 表示以 \(i\) 结尾且其中恰好有 \(j\)\(1\) 的方案数,而我们要求的是 \(G_i=\sum_{j\le i}g_{i,1}\),可以写出如下的转移方程:

  • \(a_i=0\) 时:

\[\begin{cases} g_{i,0}&=g_{i-1,0}+1\\ g_{i,1}&=g_{i-1,1}\\ G_i&=G_{i-1}+g_{i,1}=G_{i-1}+g_{i-1,1} \end{cases} \]

  • \(a_i=1\) 时:

\[\begin{cases} g_{i,0}&=0\\ g_{i,1}&=g_{i-1,0}+1\\ G_i&=G_{i-1}+g_{i,1}=G_{i-1}+g_{i-1,0}+1 \end{cases} \]

把这两个转移写成矩阵的形式就是:

  • \(a_i=0\) 时:

\[\begin{bmatrix} g_{i-1,0}\\ g_{i-1,1}\\ G_i\\ 1 \end{bmatrix} \times \begin{bmatrix} 1&0&0&0\\ 0&1&1&0\\ 0&0&1&0\\ 0&0&0&1 \end{bmatrix} = \begin{bmatrix} g_{i-1,0}+1\\ g_{i-1,1}\\ G_{i-1}+g_{i-1,1}\\ 1 \end{bmatrix} \]

  • \(a_i=1\) 时:

\[\begin{bmatrix} g_{i-1,0}\\ g_{i-1,1}\\ G_i\\ 1 \end{bmatrix} \times \begin{bmatrix} 0&1&1&0\\ 0&0&0&0\\ 0&0&1&0\\ 0&1&1&1 \end{bmatrix} = \begin{bmatrix} 0\\ g_{i-1,0}+1\\ G_{i-1}+g_{i-1,0}+1\\ 1 \end{bmatrix} \]

初始向量为 \(\begin{bmatrix}0&0&0&1\end{bmatrix}\)

因此这两个条件可以看成最标准的动态 dp 的形式,用线段树维护单点修改和区间求矩阵积两种操作即可

接下来考虑第三个条件,这个条件的分析就很简单了:

  • \(S\) 中只有一个 \(1\):直接用树状数组维护一下区间内 \(1\) 的个数即可统计
  • \(S\) 中有 \(1\)\(1\)\(1\)\(0\):对于每一对 \((a_{i-1},a_i)\),维护 \([a_{i-1}\ne a_i]\) 作为 \(i\) 的贡献,查询只需要查询 \((l,r]\) 范围内的贡献和即可

综上所述,我们可以把所有的操作用线段树或树状数组来维护

时间复杂度 \(\Theta(\delta^3(n+q)\log n)\),其中 \(\delta\) 为动态 dp 中转移矩阵的大小,此处 \(\delta=6\)

Submission Link


D. [CCC2018] 最大战略储备

Problem Link

首先把每个城市看成一个节点,那么所有的节点可以看成一个矩阵的形式,每个城市可以用一个二元组表示 \((i,j)\),其中同一个星球的城市在同一行(\(i\) 相同),在星球中的相对标号相同的城市在同一列(\(j\) 相同),那么原图中的每条边一定是形如如下两种边之一:

  • 对于某两行 \(x,y\),对于所有列 \(j\),连接 \((x,j),(y,j)\),称作“竖边”
  • 对于某两列 \(x,y\),对于所有行 \(j\),连接 \((i,x),(i,y)\),称作“横边”

显然原问题等价于在矩阵中求最小生成树

考虑继续用 Kruskal 算法,按边权将所有的边从小到大排序,手动模拟几次的过程:

  1. 假设第一条边是横边 \((x_1,y_1)\),那么可以直接连接所有的 \((x_1,j),(y_1,j)\)
  2. 假设第二条边也是横边 \((x_2,y_2)\),那么根据 Kruskal 的定义,假如 \((x_2,y_2)\) 所在的列不连通就可以连接所有的 \((x_2,j),(y_2,j)\)
  3. 假设第三条边是竖边 \((x_3,y_3)\),我们需要对所有 \((i,x_3),(i,y_3)\) 连边,但是由于有一些竖边可能会和横边构成环(例如 \((x_1,y_3)\to(x_1,y_3)\to (y_1,y_3)\to (y_1,x_3)\to(x_1,x_3)\)),观察后可以发现,每一行中的连通块,我们只需要挑一个点连接竖边
  4. 假设第四条边是横边 \((x_4,y_4)\),首先我们要判断 \((x_4,y_4)\) 所在的列是否联通,然后我们要注意 \(x_4\)\(y_4\) 列中同样的连通块只要连一个,因此这就和每一列的联通情况有关,容易想到一个断言:每一列的联通情况都是一样的,这是显然的,只要观察到:在第三次的过程中,没连接的竖边实际上的两个端点也是联通的,因此我们可以在此时将这条竖边看做联通的,因此对于每个列的联通情况只需要一个并查集就可以维护了

每次加入新横边的时候,连边的数量就是列的连通块数量,然后再连接对应的两个列,加入横边同理

因此我们只需要对于行和列分别维护并查集即可维护答案

时间复杂度 \(\Theta((p+q)\log(p+q))\)

Submission Link


E. [HNOI/AHOI2018] 寻宝游戏

Problem Link

首先对每个二进制位分别考虑,要注意到如下的观察

  • \(0\oper{and}1=0,1\oper{and}1=1\) 以及 \(0\oper{or}0=0,1\oper{or}0=1\),即 \(\oper{and}1\)\(\oper{or} 0\) 操作不影响原数的值
  • \(0\oper{and}0=0,1\oper{and}0=0\) 以及 \(0\oper{or}1=1,1\oper{or}1=1\),即 \(\oper{and}0\)\(\oper{or} 1\) 会将数值直接推平成操作数的值

显然,对于每个位最终的结果,一定是只由最后一个 \(\oper{and}0\)\(\oper{or}1\) 操作确定的

我们有这样的一个构造:把一个长度为 \(n\) 的操作序列看成 01 序列 \(\seq O\),其中 \(\oper{and}\to 1,\oper{or}\to 0\)

考虑某一位 \(i\) 上的 \(n\) 个数从小到大顺次构成的数值序列 \(\seq D_i\),我们把操作序列和数值序列取反,发现答案的第 \(i\)\(d_i=[\seq O<\seq D_i]\),其中 \(<\) 表示比较字典序

事实上,这个观察很好理解,从后往前逐位考虑,如果最高位不同,那么 \(\seq O<\seq D_i\) 就代表最后一组操作是 \(\oper{or}1\),此时 \(d_i=1\),否则代表 \(\oper{and}0\),此时 \(d_i=0\),因此此时 \(d_i=[\seq O<\seq D_i]\) 成立

如果最高位相同,那么比较次高位,等价于最后一组操作是 \(\oper{and}1\)\(\oper{or}0\),这并不改变 \(d_i\),因此结果依赖于倒数第二次操作后的 \(d_i\),等价于比较次高位,如果最终 \(\seq O=\seq D_i\),那么表示全程没有修改,\(d_i\) 直接为初值也就是 \(0\),简单做数学归纳法即可证明这个命题的正确性

我们先预处理出每个位 \(i\)\(n\) 个数从大到小逆序排列所对应的数值序列 \(\seq D_i\),对于当前 \(r\) 的某个数位 \(i\)\(r_i=0\) 表示 \(\seq D_i\le \seq O\),否则表示 \(\seq D_i>\seq O\),因此 \(\seq O\) 可以表示成 \(\seq O\in[\seq D_l,D_r)\) 的形式

注意到预处理出 \(\seq D_i\) 按字典序的排名以及 \(\seq D_i\) 作为二进制数所表示的对应的实际数值 \(\oper{int}(\seq D_i)\),那么答案就是 \(\oper{int}(\seq D_r)-\oper{int}(\seq D_l)\),可以直接快速计算(注意 \(\seq D_r<\seq D_l\) 时答案为 \(0\)

注意在无下界的情况下用全 \(0\) 串代替 \(\seq D_l\),在无上界的情况下用 \(2^{n}\),即 \(\left(1\underbrace{000\cdots 0}_{n\text{ times}}\right)\) 代替 \(\mathcal D_r\)

时间复杂度 \(\Theta(nm\log m+qm)\)

Submission Link


F. [JXOI2018] 守卫

Problem Link

考虑区间 dp:用 \(dp_{l,r}\) 表示查询区间 \([l,r]\) 的答案,我们有一个显然的观察:所有保镖覆盖的位置一定小于其所在的位置,那么 \(r\) 一定在方案中

因此考虑 \([l,r]\)\(r\) 看到的最左边的点 \(p\),可以证明:为了覆盖 \(p-1\),在位置 \(p\) 和位置 \(p-1\) 上至少有一个位置有保镖

那么可以按照 \([l,p]\)(或 \([l,p)\))和 \((p,r]\) 分裂成两段,能够证分裂出来的两段之间不会有位置存在相互看见的关系,因此答案就是 \(\min(dp_{l,p-1},dp_{l,p})+dp_{p+1,r}\)

枚举右端点,左端点从大到小移动并维护 \(p\) 即可

时间复杂度 \(\Theta(n^2)\)

Submission Link


G. [JSOI2018] 列队

Problem Link

容易证明的贪心是:当编号在 \([l,r]\) 之间的学生列队之后,他们之间左右的相对顺序是不会改变的

因此我们可以用 \(p_i\) 表示第 \(i\) 个人在 \([k,k+r-l]\) 中排在第 \(p_i\) 位,答案就是 \(\sum_{l\le i\le r} |p_i-a_i|\),考虑把绝对值拆开,答案变成 \(\sum_{p_i>a_i} p_i-\sum_{p_i>a_i}a_i+\sum_{p_i\le a_i} a_i-\sum_{p_i\le a_i} p_i\)

注意到 \(\sum p_i\) 是一段连续自然数,并且最小值(或最大值)已知,只需要知道满足条件的 \(i\) 的个数即可,第二个问题也非常简单,建立主席树即可解决

问题只剩下找到 \(a_i\) 的临界值,显然,我们在主席树上二分,通过统计某一个值域段中的 \(a_i\) 的数量来判断 \(p_i\) 的位置并决定下一步递归左子树还是右子树,查询答案时也只需要用给定的值域在主席树上求出 \(a_i\) 的数量和 \(a_i\) 的和即可

时间复杂度 \(\Theta((n+m)\log n)\)

Submission Link


H. [eJOI2019] 塔

Problem Link

首先假设我们每次操作的都是之前的整个区间,容易发现生成的序列为 \(1,1,2,4,8,\dots\)

根据值域的放缩,假设操作次数为 \(k\),容易发现一个 \(k\) 的下界是 \(\lceil\log_2 q\rceil +2\),考虑如何构造一个满足这样下界的操作序列

注意到当我们在加入第 \(i\) 个数字时,选择的区间从 \([1,i)\) 变成 \([2,i)\),那么 \(a_i\gets a_i-1\),对后续的影响为 \(-1,-1,-2,-4,\dots\) 那么 \(a_i\) 对最后一位产生 \(-2^{k-i-1}\) 的影响

那么我们修改 \(a_1\sim a_{k-1}\) 最终对答案的贡献就是 \(2^0\sim 2^{k-1}\),显然足够凑出 \(2^{\lceil \log_2q\rceil}-q\) 的值

故直接对于每一个位判断从 \(1\) 还是 \(2\) 开始即可

时间复杂度 \(\Theta(T\log q)\)

Submission Link


I. [JXOI2018] 排序问题

Problem Link

首先转化题目中的条件,假设我们有一个长度为 \(n\) 的序列,且值 \(i\) 的出现次数为 \(c_i\),那么合法的序列应该有 \(\prod_i c_i!\) 种,因此单次排序成功的概率为 \(\dfrac{\prod_ic_i!}{n!}\),最终成功的期望次数为 \(\dfrac{n!}{\prod_ic_i!}=\binom{n}{c_1,c_2,\dots}\)

由于 \(n\) 给定,我们的目标就是最小化 \(\prod_i c_i!\),显然不在 \([l,r]\) 内的 \(i\) 对答案的贡献是确定,因此只需要最小化 \(\prod _{l\le i\le r}c_i!\) 的值,又由于 \(\sum_{l\le i\le r}c_i\) 是定值,因此我们只需要让所有的 \(c_i\) 尽可能平均即可,显然可以二分这个平均值,然后暴力判断即可

预处理出所有的阶乘即可解决

\(V\)\(n+m\) 的值域,\(w\)\(r-l+1\) 的值域,时间复杂度为 \(\Theta(V+n\log w)\)

Submission Link


J. [八省联考 2018] 劈配

Problem Link

首先匹配问题第一步显然想到网络流:从源点向所有选手 \(i\) 连容量为 \(1\) 的边,从所有老师 \(j\) 向汇点连容量为 \(b_j\) 的边

匹配每个人最终志愿的时候考虑增量算法:对于每次新增选手 \(i\),从小到大枚举他的每一档志愿 \(j\),每次将 \(i\) 与他的 \(j\) 档志愿中的老师连接一条容量为 \(1\) 的边,如果在当前的图中搜出增广路,立刻停止,并找用 EK 算法一条增广路流向汇点,否则删除这些边,考虑 \(j+1\) 档志愿

第二问也类似,在匹配 \(i\) 之前,对于所有 \(\ge i\) 的人 \(j\) 判断当他们此时连接 \(1\sim s_k\) 档志愿能否得到增广路,如果能就用 \(j-i\) 更新答案,注意此时的所有边都要删除,也不要用 EK 增广

单次搜索找增广路的时间复杂度是 \(\Theta(nC)\) 的,对于第一问,每个人要找 \(\Theta(m)\) 次增广路,对于第二问,每次有 \(\Theta(n)\) 个人要找增广路,因此时间复杂度是 \(\Theta(n^2(n+m)C)\)

Submission Link


K. [APIO2020] 有趣的旅途

Problem Link

注意到 \(\oper{dist}(u,v)\)\(\oper{LCA}(u,v)\) 有关系,比较不好处理,因此考虑定根,即选定一个点 \(\oper{root}\) 作为根,使得所有 \(\oper{LCA}(p_{i-1},p_i)=\oper{root}\)

注意到此时我们选择的相邻的 \(p_i\) 就需要来自 \(\oper{root}\) 的不同的两棵子树,因此一个自然的想法是让 \(\oper{root}\) 的各子树大小尽可能接近

为了让子树大小尽可能接近,我们想到了树上的一个特殊点——重心,容易发现,重心在这里具有非常好的性质:当 \(\oper{root}\) 只有两棵子树 \(T_A,T_B\) 时,\(|T_A|+|T_B|\le \left\lfloor\dfrac n2\right\rfloor,|T_A|+|T_B|=n-1\),我们能发现此时 \((|T_A|,|T_B|)\) 的差至多为 \(1\),此时直接交替选择 \(T_A,T_B\) 中深度最大的点即可

接下来我们需要处理 \(\oper{root}\) 有三棵子树 \(T_A,T_B,T_C\) 的情况,首先依然想到承袭上面的思路,每次选择上一次未被选择的两棵子树中深度最大的节点,不过这个做法可能有问题,当某个时刻有两棵子树都被删空之后,我们将无法继续进行这个操作

注意到如下的观察:

当我们不断执行如上操作并且最终会剩下一棵子树时,在这个过程中,总存在某个时刻,使得 \(\oper{root}\) 的某棵子树大小恰好等于另外两棵子树大小之和

假如这个观察成立,那么我们可以在在这个时刻把这两棵子树合并,转化为两棵子树的情况,不过一种需要特殊注意的情况:假设当前 \(|T_A|=|T_B|+|T_C|\),并且我们上一次选的节点是 \(T_B\),且此时 \(T_C\) 子树的最大深度大于上一个节点的深度,正常情况下我们会选择 \(T_C\),但在合并 \(T_B,T_C\) 之后,我们会选择 \(T_A\),此时要先选择 \(T_C\) 中深度最大的节点再操作才可保证正确性

下面给出这个观察的证明:

证:

对于某个时刻,我们定义函数 \(f(T_A,T_B,T_C)=|T_A|+|T_B|+|T_C|-2\times\max\{|T_A|,|T_B|,|T_C|\}\)

显然,最开始由于 \(\oper{root}\) 是重心,因此 \(\max\{|T_A|,|T_B|,|T_C|\}\le\left\lfloor\dfrac n2\right\rfloor\),因此 \(f(T_A,T_B,T_C)\ge (n-1)-2\times\left\lfloor\dfrac n2\right\rfloor=-1\)

而我们要证明的是存在某个时刻 \(f(T_A,T_B,T_C)=0\),首先,当 \(f(T_A,T_B,T_C)=-1\) 时,直接操作大小最大的子树即可

在剩下的情况中, \(f(T_A,T_B,T_C)\) 的变化量 \(\Delta\) 分为以下几种:

  • 操作的子树 \(T\) 是最大子树,操作完还是最大子树,此时 \(\Delta=1\)
  • 操作的子树 \(T\) 是最大子树,操作完不是最大子树,此时 \(\Delta=-1\)
  • 操作的子树 \(T\) 不是最大子树,此时 \(\Delta=-1\)

综上,\(f(T_A,T_B,T_C)\) 在整值值域上连续,而当最终我们留下一棵子树 \(T\) 不删空时,一定有 \(f(T_A,T_B,T_C)=-|T|<0\),因此 \(f(T_A,T_B,T_C)\) 从正值变化到负值的过程中一定会经过 \(f(T_A,T_B,T_C)=0\),即某个子树大小等于另外两个子树大小之和的情况

综上所述,对于 \(T_A,T_B,T_C\) 排序后就可以解决这个问题

接下来考虑如何利用交互库:

  • 首先求出树的重心,我们可以先随意指定一个根(假设为 \(0\)),然后调用 attractionsBehind(0,i) 得到此时 \(i\) 的子树大小 \(siz_i\),找到 \(n-siz_i\le \left\lfloor\dfrac n2\right\rfloor\)\(siz_i\) 最小的 \(i\) 即为重心
  • 求出 \(\oper{root}\) 的儿子 \(A,B,C\),显然调用 hoursRequired(root,i) 得到每个点的深度 \(dep_i\)\(dep_i=1\) 的即为 \(\oper{root}\) 的儿子
  • 将所有顶点划分到 \(T_A,T_B,T_C\) 中,调用 hoursRequired(A,i)hoursRequired(B,i) 求出 \(\oper{dist}(A,i),\oper{dist}(B,i)\),设 \(i\)\(X\) 的子树里,那么显然 \(\oper{dist}(X,i)+1=dep_i\),且 \(A,B,C\) 中这样的 \(X\) 有且仅有一个,故判断 \(A,B\) 是否满足,若均不满足则 \(X=C\)

综上,我们的总操作次数是 \(4\times n\) 的,可以通过本题

时间复杂度 \(\Theta(n\log n)\)

Submission Link


L. [ZJOI2018] 胖

Problem Link

\(dis_i\) 表示 \(1\)\(i\) 的距离,那么任意两个点 \(i,j\) 的距离是 \(|dis_i-dis_j|\)

考虑某个 \(a_i\) 能够更新的范围,对于某个 \(x\)(假设 \(x\le a_i\)),\(a_i\) 能够更新 \(x\) 当且仅当 \([2\times x-a_i,a_i)\) 中不包含比 \(a_i\) 更优的决策点,即:

  • 对于所有 \(a_j\in[2\times x-a_i,x)\),有 \(c_j+dis_x-dis_{a_j}<c_i+dis_{a_i}-dis_x\) 成立
  • 对于所有 \(a_j\in[x,a_i)\),有 \(c_j+dis_{a_j}-x<c_i+dis_{a_i}-dis_x\) 成立

注意到 \(x\) 显然有可二分性,因此二分 \(x\),检验的时候用 ST 表维护对应两个区间中的 \(c_j\pm dis_{a_j}\) 最小值并比较检验即可,注意 ST 表每次只维护给定的 \(m\) 个元素,因此在 ST 表上查询的对应区间需要通过二分得到

类似地可以二分出 \(a_i\) 可以更新的区间的右端点,因此枚举每个 \(i\) 二分得到其有贡献的区间并求和即可得到答案

特别地,当某两个 \(a_i,a_j\) 更新 \(x\) 的代价相同且 \(|x-a_i|=|x-a_j|\) 时,我们会漏统计此时的贡献,在二分左端点或右端点其中之一时检验时特判这种情况并判为可行即可

时间复杂度 \(\Theta(n+K\log K\log n)\)

Submission Link


M. [HAOI2018] 染色

Problem Link

首先可以先解出:有 \(\ge k\) 种颜色恰好出现了 \(s\) 次的方案数 \(f_k\)

\[f_k=\binom mk\times \dfrac{n!}{(n-s\times k)!\times(s!)^i}\times (n-s\times k)^{m-k} \]

假设恰好有 \(k\) 种颜色出现而恰好 \(s\) 次的方案数为 \(g_k\),根据广义容斥原理,我们有:

\[g_k=\sum_{j=k}^m (-1)^{j-k}\binom jk f_j \]

对组合数进行展开并且稍作变形可以得到:

\[\begin{aligned} g_k&= \sum_{j=k}^m (-1)^{j-k}\binom jk f_j\\[2ex] &=\sum_{j=k}^m (-1)^{j-k} \dfrac{j!}{(j-k)!\times k!} \times f_j\\[2ex] &=\dfrac 1 {k!}\sum_{j=k}^m\dfrac{(-1)^{j-k}}{(j-k)!}\times(j!\times f_j) \end{aligned} \]

注意到求和式里面的东西只和 \(j-k,j\) 有关,设 \(a_k=\dfrac{(-1)^{k}}{k!},b_k=k!\times f_k\),那么原式可以进一步化简成:

\[g_k=\dfrac 1{k!}\sum_{i=0}^m\sum_{j=0}^m[i-j=k]\times b_i\times a_j \]

注意到两个变量的差是定值,因此容易想到将一个变量取反使得和变为定值,即令 \(b'_k=b_{m-k}\),那么有:

\[g_k=\dfrac{1}{k!}\sum_{i=0}^m\sum_{j=0}^m[i+j=m-k]\times b_i\times a_j \]

此时可以发现这是一个卷积的形式,因此 \(g_{m-k}\) 的生成函数可以由 \(b_{m-k}\) 的生成函数和 \(a_k\) 的生成函数卷积得来,一次 NTT 即可求出所有 \(g_k\),统计贡献并求和即可

时间复杂度 \(\Theta(n+m\log m)\)

Submission Link


N. [CQOI2018] 破解D-H协议

Problem Link

用 BSGS 解出满足 \(g^x\equiv A\pmod P\)\(x\),答案即为 \(B^x\bmod P\)

时间复杂度 \(\Theta(n\sqrt P\log P)\)(BSGS 内部用哈希表存储可以做到 \(\Theta(n\sqrt P)\)

Submission Link


O. [BalticOI2019] 潜艇

Problem Link

bitset 维护每个位置当前是否可以作为结束位置,用左右移位模拟潜艇向四个方向移动的过程,注意在移动之前要把第一列/第一行/最后一列/最后一列的位置设为不可达再移动,不然可能会越界

时间复杂度 \(\Theta\left(M\times\frac{RC}{\omega}\right)\)

Submission Link


P. [CQOI2018] 九连环

Problem Link

\(f_i\) 表示拆下 \(i\) 个环的最小步数,\(g_i\) 为装上 \(i\) 个环的最小步数

考虑 \(f_n\) 求解的过程:首先我们要拆下前 \(n-2\) 个环,然后拆下第 \(n\) 个环,然后装上前 \(n-2\) 个环并拆下前 \(n-1\) 个环,因此有 \(f_n=f_{n-2}+1+g_{n-2}+f_{n-1}\),同理也可得 \(g_n=g_{n-1}+f_{n-2}+1+g_{n-2}\),且 \(f_0=g_0=0,f_1=g_1=1\),根据归纳法我们知道 \(f_n=g_n\),因此我们只需要解递推式 \(f_n=f_{n-1}+2f_{n-2}+1\) 即可,其中 \(f_0=0,f_1=1\)

显然对 \(f_i\) 建立生成函数 \(F(x)\),有如下推导:

\[\begin{aligned} F(x) &=\sum_{i=0}^{\infty} f_i\times x^i\\ &=f_0\times x^0+f_1\times x^1+\sum_{i=2}^\infty f_i\times x^i\\ &=x+\sum_{i=2}^\infty x^i(f_{i-1}+2f_{i-2}+1)\\ &=x+x\left(\sum_{i=1}^\infty x^i\times f_i\right)+2x^2\left(\sum_{i=0}^\infty x^i\times f_i\right)+\sum_{i=2}^{\infty} x^i\\ &=\sum_{i=1}^\infty x^i+xF(x)+2x^2F(x)\\ &=\dfrac x{1-x}+xF(x)+2x^2F(x)\\ &=\dfrac{x}{(1-x)(1-x-2x^2)}\\ &=\dfrac{x}{(1-x)(1+x)(1-2x)}\\ &=\dfrac{-\frac 12}{1-x}+\dfrac{-\frac 16}{1+x}+\dfrac{\frac 23}{1-2x}\\ &=\dfrac16\sum_{i=0}^\infty [2^{2n+2}-3-(-1)^i]x^i \end{aligned} \]

因此我们得到 \(f_i=\dfrac{2^{2n+2}-3-(-1)^i}{6}\),用 FFT 优化快速幂可以通过本题,时间复杂度 \(\Theta(mn\log^2n)\)

事实上,我们用压位高精也能通过本题,并且在压位较多时有较优秀的表现

时间复杂度 \(\Theta(mL^2\log n)\),其中 \(L\) 表示答案的位数,在压 \(9\) 位高精时,\(L\approx3000\),可以通过本题

Submission Link


Q. [WC2018] 州区划分

Problem Link

显然考虑状压 dp,用 \(\m S\) 表示状压点集,\(dp_{\m S}\) 表示划分点集 \(\m S\) 的答案,用 \(\oper{valid}(\m S)\) 表示 \(\m S\) 表示的点集可以划分在一个州区,即 \(\m S\) 这个点集的导出子图不存在欧拉回路(注意要同时判断连通性和度数)用 \(\oper{cost}(\m S)\) 表示 \(\m S\) 中的节点 \(w\) 值之和,状态转移方程如下:

\[dp_{\m S}=\sum_{\m T\subseteq \m S} [\oper{valid}(\m T)]\times dp_{\m S-\m T}\times\dfrac{\oper{cost}(\m T)}{\oper{cost}(\m S)} \]

其中边界条件为 \(dp_{\varnothing}=1\),稍作变化可以写成下式:

\[dp_{\m S}=\dfrac1{\oper{cost}(\m S)}\sum_{\m A,\m B}[\m A\cup\m B=\m S]\times[\m A\cap\m B=\varnothing]\times([\oper{valid}(\m A)]\times\oper{cost}(\m A))\times dp_{\m B} \]

注意到 \(\m A\cup\m B=\m S\) 是一个标准的或运算 FWT 的形式,因此我们考虑用 FWT 优化转移的复杂度,注意到比较难处理的是 \(\m A\cap\m B=\varnothing\),不过我们可以考虑用 \(|\m A|+|\m B|=|\m S|\) 来约束,也就是用 \(dp_{|\m S|,\m S}\) 来表示状态,转移 \(dp_k\) 的时候从所有 \(i+j=k\)\(dp_i\otimes cost_j\) 处转移,\(\otimes\) 表示位运算卷积

时间复杂度 \(\Theta((n^2+m)2^n)\)

Submission Link


R. [省选联考 2021 B 卷] 取模

Problem Link

先考虑对于固定的模数 \(p\) 如何解决该问题,令 \(r_i=a_i\bmod p\)\((r_i+r_j)\bmod p\) 的值有两种可能:\(r_i+r_j\)\(r_i+r_j-p\),对于第二种情况,直接选择最大和次大的 \(r_i,r_j\) 即可,对于第一种情况,我们用双指针或二分对每个 \(r_i\) 找到小于 \(p-r_i\) 的最大 \(r_j\) 并统计即可

注意到这样的单次操作时间复杂度是 \(\Theta(n\log n)\) 的,考虑剪枝,两个显然的剪枝分别是:

  • 从大到小考虑模数 \(a_i\),当当前 \(a_i\) 不超过答案时,直接停止更新
  • 对于多个相同的 \(a_i\) 只考虑一个

在使用了这两个优化之后,枚举次数变成 \(\Theta(\log w)\) 级别,其中 \(w\)\(a_i\) 的值域,证明如下:

证:

假设在求出答案之前,枚举过的模数从小到大依次是 \(\oper{ans}<p_1<p_2<\cdots<p_k\),那么根据第一个剪枝,显然 \(p_{i-1}+p_{i-2}\ge p_i\),否则 \(p_{i-1}+p_{i-2}<p_i\),因此 \(\oper{ans}>p_{i-1}+p_{i-2}>p_{i-1}>p_{i-2}\) 与第一个剪枝矛盾

因此我们得到 \(ans\ge (p_{i-1}+p_{i-2})\bmod p_i\ge p_{i-1}+p_{i-2}-p_i\),从而推出 \((p_i-\oper{ans})>(p_{i-1}-\oper{ans})+(p_{i-2}-\oper{ans})\)

由于上式我们可以得到 \(p_1-\oper{ans}\sim p_k-\oper{ans}\) 的增长速度超过斐波那契数列,由于比内公式我们知道斐波那契数量成指数级增长,因此 \(k\) 的个数是 \(\Theta(\log w)\) 级别的,原命题得证

时间复杂度 \(\Theta(n\log n\log w)\)

Submission Link


S. [CTSC2018] 假面

Problem Link

考虑先处理第一个操作,显然用 \(dp_{i,j}\) 表示第 \(i\) 个人当前剩余生命值为 \(j\) 的概率,显然以 \(p\) 的命中概率每次攻击某个人 \(i\),有转移 \(dp_{i,j}\gets (1-p)\times dp_{i,j}+p\times dp_{i,j+1}\),特别地,\(dp_{i,0}\gets dp_{i,0}+p\times dp_{i,1}\)

显然在已知 \(dp_{i,j}\) 的基础上,求出最终每个人的剩余生命的期望也可以轻松求出

接下来只需要解决技能命中的问题,显然我们只关心一个人是否存活,记第 \(id_i\) 个人存活的概率为 \(p_i\),显然 \(p_i=1-dp_{i,0}\),接下来我们需要对于每个 \(id_i\),计算除了 \(id_i\) 之外恰好有 \(0,1,2,\dots,k\) 个人存活的概率,显然可以看成一个有关存活人数的背包

看到背包问题显然想到多项式优化,定义多项式 \(F_i(x)=\prod_{j\ne i} p_{j}x+(1-p_{j})\),我们只需要求出所有 \(F_i(x)\) 即可,观察到这是一个回撤背包的形式,先求出 \(F(x)=\prod_j p_{j}x+(1-p_j)\),每次计算 \(i\) 的时候只需要用 \(F(x)\) 除以 \(p_ix+(1-p_i)\) 即可得到 \(F_i(x)\),这两步的卷积可以直接暴力解决,注意当撤回贡献时若存在 \(p_i=0\)\(1-p_i=0\) 的特殊情况可能要特判

时间复杂度 \(\Theta(Cn^2+Qw)\),其中 \(w\)\(m_i\) 的值域

Submission Link


T. [NOI2018] 归程

Problem Link

在使用车之后,Yazid 可以从 \(u\) 通过无积水边到达的整个连通块的任意一个点开始,步行前往 \(1\),因此答案是这个连通块中的点到 \(1\) 的距离最小值,先用 Dijkstra 预处理一遍所有的 \(\oper{dist}(1,i)\) 即可快速求解

接下来只需要维护从 \(u\) 出发,经过海拔 \(>h\) 的边能到的连通块,显然考虑 Kruskal 重构树,边权按海拔从大到小排序,建出 Kruskal 重构树,在重构树上倍增,求出权值 \(>h\) 的最高点,其子树对应的连通块即为所求,而快速维护对应子树的最小值,不需要 RMQ,只需要在 Kruskal 合并两棵子树的时候把对应子树的最小距离也一起合并到父亲上即可

时间复杂度 \(\Theta(m\log m+q\log n)\)

Submission Link


U. [HNOI/AHOI2018] 游戏

Problem Link

显然每个点 \(i\) 能到达的是一段连续的区间 \([l_i,r_i]\),回答询问时只需要判断 \(t\in[l_s,r_s]\) 是否成立即可

我们考虑最暴力的拓展,假如可以向左走就向左拓展,可以向右走也一样向右拓展,注意到每个位置向左走之后的节点在拓展完之后可以记忆化,即当 \(i\to i-1\) 可行时,\(l_{i-1}\le l_i\) 一定成立,\(l_i\) 可以直接从 \(l_{i-1}-1\) 去拓展,注意到 \(l_i\) 的指针是不断左移的,因此可以证明时间复杂度是线性的,同理,向右拓展的时间复杂度也是线性的

综上所述,我们只需要记忆化并暴力搜索拓展即可

时间复杂度 \(\Theta(n)\)

Submission Link


V. [HNOI/AHOI2018] 排列

Problem Link

首先我们有一个观察:某个 \(a_i\) 表示在 \(p\) 中,\(a_i\) 一定要在 \(i\) 之前出现,因此连接所有 \(i\to a_i\),形成的应该是一棵以 \(0\) 为根的树,否则显然无解

接下来考虑 \(1\) 的位置,显然的观察是 \(1\) 一定要尽可能早选,也就是当 \(1\) 的父亲被选择之后一定要立刻选 \(1\)

因此我们可以把 \(1\)\(1\) 的父亲看成一个连接的序列,这样我们每次都相当于在整个树中找一个价值最小的序列把这个序列和其父亲连接起来

接下来我们要判断哪个序列价值最小,假设一个序列 \(x\) 中节点的数量为 \(siz_x\),权值和为 \(w_x\),那么对于两个序列 \(a,b\)\(w_{ab}-w_{ba}=w_b\times siz_a-w_a\times siz_b\),因此 \(a\)\(b\) 先被取出合并当且仅当 \(w_a\times siz_b<w_b\times siz_a\),注意到这实际上等价于 \(\dfrac{w_a}{siz_a}<\dfrac{w_b}{siz_b}\),即比较平均值,因此这个比较是全序的,直接用 set 维护所有节点即可

时间复杂度 \(\Theta(n\log n)\)

Submission Link


W. [IOI2020] 连接擎天树

Problem Link

注意到 \(p_{i,j}\in\{0,1,2\}\) 时很好解决,图一定是一个基环树森林的形式:

  • \(p_{i,j}=0\),则表示 \(i,j\) 不在同一棵基环树中
  • \(p_{i,j}=1\),则表示 \(i,j\) 在同一棵基环树的同一棵子树中
  • \(p_{i,j}=2\),则表示 \(i,j\) 在同一棵基环树的不同子树中

接下来考虑 \(p_{i,j}=3\) 的情况,显然至少要在基环树的基础上添加一条边,那么至少会额外形成一个环,可以发现,在有两个环的基环树中,我们总能找到一组点对 \((u,v)\) 满足 \(u\to v\) 的路径有 \(4\) 条,因此我们可以断言:存在 \(p_{i,j}=3\) 时原问题无解

接下来考虑如何实现代码,首先判断 \(p_{i,j}=3\) 的情况,然后用并查集处理 \(p_{i,j}=0\) 的情况,把 \(p_{i,j}\ne 0\) 的加入一个连通块,并判断同个联通块是否有 \(p_{i,j}=0\) 的情况,如果有则返回无解

接下来考虑 \(p_{i,j}=2\) 的情况,重新初始化并查集,把 \(p_{i,j}=1\) 的加入同一个连通块,并判断同个连通块是否有 \(p_{i,j}=2\) 的情况,如果有则返回无解,然后把同一个连通块的点连成链,把不同连通块的代表元提取出来连成一个环,需要特判环的大小 \(=2\) 的情况,此时若形成环必然需要重边,不符合题意,直接报告无解即可

时间复杂度 \(\Theta(n^2\alpha(n))\)

Submission Link


X. [eJOI2019] 挂架

Problem Link

首先将挂钩的下标转成二进制下的 \(0\sim 2^n-1\),并模拟 \(n=3\) 时的操作顺序:\(000\to100\to010\to110\to001\to101\to011\to111\),我们可以看到,最高位以 \(2^0\) 为周期切换 \(0\)\(1\),次高位以 \(2^1\) 为周期切换 \(0\)\(1\),其他位同理

事实上,最高位每次切换相当于维持第一层的 \(2^0\) 根平衡杆平衡,次高位每次切换相当于维持第二层的 \(2^1\) 根平衡杆平衡,其他位同理,因此答案的每一位上是 \(0\) 还是 \(1\) 可以直接取出其对应位的值就可以确定,最后再在 \(\bmod 10^9+7\) 意义下逐位还原答案即可,注意处理时下标是 \(0\) 开始标号的,因此读入的 \(k\) 要减一,输出的答案要加一

时间复杂度 \(\Theta(n+\log k)\)

Submission Link


Y. [WC2021] 括号路径

Problem Link

定义 \(\oper{valid}(u,v)\) 表示 \(u\to v\) 是一条合法括号路径,则如下的结论是显然的:

  • 双向性:\(\oper{valid}(u,v)\iff\oper{valid}(v,u)\)
  • 可传递性:\(\oper{valid}(u,v)\land\oper{valid}(v,w)\implies \oper{valid}(u,w)\)

因此我们可以把 \(\oper{valid}(u,v)\) 想象成 \(u,v\) 的连通性,用并查集维护,而每次拓展时只需要找到两条 \((x,u,w),(x,v,w)\) 的边即可把 \(u,v\) 缩成一个点,先用 map 维护每个点每种颜色的出边,合并用 BFS 逐步维护即可

每次合并两个点 \(u,v\) 时维护一下出边,把能合并的 \((u,u',w),(v,v',w)\) 也合并起来,这一步需要用启发式合并来保证复杂度,每次每个点的出边集至少扩大两倍,这样总体复杂度是 \(\Theta(m\log m)\) 的,足够通过本题

时间复杂度 \(\Theta(m\log k\log m)\)

Submission Link


Z. [省选联考 2021 A 卷] 矩阵游戏

Problem Link

注意到没有值域限制的问题是很显然的,直接假设 \(a_{j,m}=a_{n,i}=0\),剩下的可以从后往前逆序递推

接下来考虑解决 \([0,10^6]\) 的值域限制,注意到对于某一行或某一列,只有交替地进行 \(+x,-x\) 的操作才保证不影响答案,假设第 \(i\) 行的变化量为 \(+r_i.-r_i,+r_i,\dots\),第 \(j\) 列的变化量为 \(+c_j,-c_j,+c_j,\dots\),那么 \(a_{i,j}\) 的变化量 \(\Delta_{i,j}\) 有以下 \(4\) 种可能:

  • \(2\mid i,2\mid j\) 时:\(\Delta_{i,j}=-r_i-c_j\)
  • \(2\nmid i,2\mid j\) 时:\(\Delta_{i,j}=r_i-c_j\)
  • \(2\mid i,2\nmid j\) 时:\(\Delta_{i,j}=c_j-r_i\)
  • \(2\nmid i,2\nmid j\) 时:\(\Delta_{i,j}=r_i+c_j\)

并且我们发现限制条件可以写成 \(-a_{i,j}\le \Delta_{i,j}\le A-a_{i,j}\),其中 \(A=10^6\),因此考虑用差分约束解决此问题,不过注意到有两种情况的 \(\Delta_{i,j}\) 是和的形式,因此我们考虑对 \(2\mid i\)\(r_i\)\(2\nmid j\)\(c_j\) 取反得到 \(x_i,y_j\),得到 \(\Delta_{i,j}\) 的变化形式为:

  • \(i\bmod 2\ne j\bmod 2\) 时:\(\Delta_{i,j}=y_j-x_i\)
  • \(i\bmod 2= j\bmod 2\) 时:\(\Delta_{i,j}=x_i-y_j\)

因此我们可以把 \(x_i,y_j\) 用差分约束系统表示,用 SPFA 求解即可

时间复杂度 \(\Theta(nm(n+m))\)

Submission Link

posted @ 2023-02-23 17:24  DaiRuiChen007  阅读(96)  评论(1编辑  收藏  举报