AGC054 题解(A-C)

Solution

A

无脑 $\text{dp}$ 即可。

B

花了好长时间才做出来。

假设 $T$ 依次选择了橘子 $t_1,t_2,\cdots ,t_k$，$A$ 依次选择了橘子 $a_1,a_2,\cdots ,a_{n-k}$，考虑构造一组满足上述要求的方案。

首先，一定是 $T$ 先选，于是他先拿了 $t_1$。接下来的一个橘子，必须是 $t_2$ 或 $a_2$，但是若选择了 $t_2$，那么它就被 $A$ 拿了，与分配方案不符。于是，我们不得不选择 $a_1$。接下来，若 $T$ 的权值和大于 $A$ 的权值和，就放 $a_i$，否则放 $t_j$。只有这样，才能保证方案合法。

从而，若钦定了 $(t,a)$，那么就只有 $1$ 种选法了。更进一步的，若我们钦定了 $(t^{\prime },a^{\prime })$，其中 $t^{\prime },a^{\prime }$ 分别表示 $T,A$ 两人选出橘子的集合（没有顺序限制），那么就会有 $|t^{\prime }|!\times |a^{\prime }|!$ 的贡献。从而，答案即为，所有权值和为 $\frac{s}{2}$ 的集合大小的阶乘之和，其中 $s={\sum }_{i=1}^n{W}_i$。

于是就可以愉快地 $\text{dp}$ 啦。

令 $f_{i,j,k}$ 表示，目前看到第 $i$ 个位置，总共选了 $j$ 个，且选出的权值和为 $k$。有转移如下：

$$\begin{gathered} f_{i,j,k}\to f_{i+1,j+1,k+W_{i+1}} \\ {f}_{i,j,k}\to f_{i+1,j,k} \end{gathered}$$

答案即为

$$\sum _{i=1}^n{f}_{n,i,\frac{s}{2}}i!$$

特别的，若 $s\equiv 1(\mathrm{m}\mathrm{o}\mathrm{d}2)$，那么不执行上述 $\text{dp}$ 并直接输出 $-1$。

总复杂度 $O(n^{2}s)$，本题被解决。应该和官方题解差不多吧。

C

考场上不知道在想什么，写了一个暴力找规律然后发现了规律，结果不会表达出来，甚至写出来了一个非常复杂的式子，导致没有在考场上切掉本题。菜得真实。

算法一

考虑枚举 $P$，并判断它是否会最终变成 $P^{\prime }$。

关键在于构造一个最优方案。不难得到: 每次找到一个最小的 $i$ 使得 $[1,i-1]$ 中比 $P_i$ 大的数超过 $k$ 个且 $P_{i-1}^{\prime }>P_i^{\prime }$，并交换 $(i-1,i)$。

这个方案肯定是能够使 $P$ 最终合法的，但是它是否是最优的呢？

考虑交换次数的下界是多少。令 $cn{t}_i$ 表示，初始时 $[1,i-1]$ 中，有多少个 $j$ 使得 $P_j^{\prime }>P_i^{\prime }$。对于每次交换，至多会有一个 $cn{t}_i$ 被减去了 $1$，于是下界就是
$$\sum _{i=1}^n\max (cn{t}_i-K,0)$$

可以证明，我们构造的操作方案恰好达到了这个下界。为什么呢？因为，每操作一次，均恰好有一个 $cn{t}_i$ 减去了 $1$，且每次都能找到一个合法的位置 $i$ 并进行操作。这样一来，我们的操作方案就是最优的啦。

由于共有 $n^2$ 轮，每轮都要在 $O(n^2)$ 的代价内找到对应的 $i$ 并交换 $(i-1,i)$，所以复杂度是惊人的 $O(n!n^4)$。

算法二

现在关键在于，对于某一个初始格局 $P^{\prime }$，快速算出其最终的格局长什么样。

考虑对于某一个 $P_i^{\prime }$，它最终到了哪里。不难发现，一旦它开始向左移动，那么它就会移动到左边第 $K+1$ 个比它大的位置左边，并永远停下来。换言之，算法一中的操作流程与下面的流程是等价的。

for (int i=2;i<=n;i++){
	int p=0,cnt=0,now=i;
	//p:向左移动的次数 cnt:左边比它大的数的个数 now:无实际意义
	for (int j=1;j<i;j++){
		if (q[j]>q[i])  cnt++;
		if (cnt==k+1){p=i-j;break;}//找到第 k+1 个比它大的，需要移动到它的左边
	}
	for (int j=1;j<=p;j++)  swap(q[now],q[now-1]),now--;//进行若干次连续的移动
}

复杂度 $O(n!n^2)$。

算法三

算法一与算法二都不能在多项式时间复杂度内完成计算，但是它们能给我们很大的启发。请看下面的引理：

Lemma

对于在给定序列 $P$ 中的某个位置 $P_i$，且第 $K$ 个大于 $P_i$ 的位置为 $j$，那么 $P^{\prime }$ 中 $P_i$ 出现的位置一定在 $P_j$ 之后。

这是充分必要条件。

---

从而，我们只需要找到所有 $y_i=K$ 的位置，固定它们的相对顺序，并把其他位置给依次插入进来。根据乘法原理不难得到，答案即为每个数可以被插入的位置数量，则

$$ans=\prod _{i=1}^n[y_i=K](n-i+1)$$

采用树状数组计算出 $y$，然后扫描一遍即可计算出答案。总复杂度 $O(n\log n)$。