AT5695「AGC041D」Problem Scores

Address

• AT5695

Solution

• 尝试简化第三个限制，设 $k=\lfloor \frac{n-1}{2}\rfloor$，我们取前 $k+1$ 个数作为第一个集合，取后 $k$ 个数作为第二个集合，因为这两个集合的差是所有选集合的方案中最小的，所以只要这两个集合满足条件，所有的集合都能满足条件。

Algorithm 1

• 因为 $n$ 的奇偶性会影响中间是否会多出一个数，我们分情况讨论：

  1. 若 $n$ 是偶数，枚举第 $k+2$ 个数为 $x$，那么对剩下的数有如下要求：

     1. $x$ 与前 $k+1$ 个数的差 $\in [0,x)$ 且差递减；
     2. 后 $k$ 个数与 $x$ 的差 $\in [0,n-x]$ 且差递增；
     3. $x$ 与前 $k+1$ 个数的差的和加上后 $k$ 个数与 $x$ 的差的和小于 $x$。

     设 $f_{i,j}$ 表示已经确定了前 $i$ 个数、$x$ 与这些数的差的和为 $j$ 的方案数。

     转移即从大到小枚举差 $l(l\in [0,n))$，令 $f_{i+1,j+l}=f_{i+1,j+l}+f_{i,j}$。

     由于第二维 $j\le n$，存在合法方案时一定有 $il\le n$，时间复杂度 $\mathcal{O}(n^2\ln n)$。

     后 $k$ 个数的方案可以看做是选取若干个后缀区间 +1，设 $g_{i,j}$ 表示已经选了 $i$ 个后缀、这些后缀的长度之和为 $j$ 的方案数。

     转移即从大到小枚举后缀长度 $l(l\in [0,k])$，令 $g_{i+1,j+l}=g_{i+1,j+l}+g_{i,j}$，时间复杂度同样是 $\mathcal{O}(n^2\ln n)$。

     最后的答案即为：
     $$\begin{array}{r}\sum _{x=1}^n\sum _{i=0}^{x-1}{f}_{k+1,i}\sum _{j=0}^{x-i-1}{g}_{n-x,j}\end{array}$$
     预处理 $g$ 第二维的前缀和即可 $\mathcal{O}(n^2)$ 计算。

  2. 若 $n$ 是奇数，枚举第 $k+1$ 个数为 $x$，只要最后算答案时把 $f_{k+1,i}$ 改为 $f_{k,i}$，其它部分与偶数的情况完全相同。

• 总的时间复杂度 $\mathcal{O}(n^2\ln n)$。

Code 1

#include <bits/stdc++.h>

const int N = 5005;
int f[N][N], g[N][N];
int n, ans, mod;

inline void add(int &x, int y)
{
	x += y;
	x >= mod ? x -= mod : 0;
}

int main()
{
	scanf("%d%d", &n, &mod);	
	int k = n - 1 >> 1;
	f[0][0] = g[0][0] = 1;
	for (int c = n - 1; c >= 0; --c)
		for (int i = 0, im = !c ? n : n / c; i <= im; ++i)
			for (int j = 0; j <= n - c; ++j)
				add(f[i + 1][j + c], f[i][j]);
	for (int c = k; c >= 0; --c)
		for (int i = 0, im = !c ? n : n / c; i <= im; ++i)
			for (int j = 0; j <= n - c; ++j)
				add(g[i + 1][j + c], g[i][j]);
	for (int i = 0; i <= n; ++i)
		for (int j = 1; j <= n; ++j)
			add(g[i][j], g[i][j - 1]);
	if (!(n & 1))
	{
		for (int x = 1; x <= n; ++x)
			for (int j = 0; j < x; ++j)
				ans = (1ll * f[k + 1][j] * g[n - x][x - j - 1] + ans) % mod;
	}
	else
	{
		for (int x = 1; x <= n; ++x)
			for (int j = 0; j < x; ++j)
				ans = (1ll * f[k][j] * g[n - x][x - j - 1] + ans) % mod;
	} 
	printf("%d\n", ans);
	return 0;
}

Algorithm 2

• 考虑将 $A$ 数组差分，去掉单调性的限制，令差分数组为 $B$。

• 令 $B_0=1$，则 $A_i=1+\sum _{j=1}^i{B}_j$，因此其中一个限制条件就是 $\sum _{i=1}^n{B}_i<n$。

• 同样对 $n$ 的奇偶性进行讨论：

  1. 若 $n$ 为奇数，$\lfloor \frac{n-1}{2}\rfloor +1=\frac{n+1}{2}$，列出限制条件：
     $$\begin{array}{rl}\sum _{i=1}^{\frac{n+1}{2}}{A}_i& >\sum _{i=\frac{n+1}{2}+1}^n{A}_i\\ \sum _{i=1}^{\frac{n+1}{2}}(1+\sum _{j=1}^i{B}_j)& >\sum _{i=\frac{n+1}{2}+1}^n(1+\sum _{j=1}^i{B}_j)\\ 2\sum _{i=1}^{\frac{n+1}{2}}\sum _{j=1}^i{B}_j& \ge \sum _{i=1}^n\sum _{j=1}^i{B}_j\\ 2\sum _{i=1}^{\frac{n+1}{2}}(\frac{n+1}{2}-i+1)B_i& \ge \sum _{i=1}^n(n-i+1)B_i\end{array}$$
     简单移项后，我们可以得到：
     $$\begin{array}{r}\sum _{i=1}^n{C}_i{B}_i\le 0\end{array}$$
     其中：
     $$C_i=\{\begin{array}{ll}i-2& 1\le i\le \frac{n+1}{2}\\ n-i+1& \frac{n+1}{2}<i\le n\end{array}$$

  2. 若 $n$ 为偶数，同样可以推出 $C_i$ 的表达式：
     $$C_i=\{\begin{array}{ll}i-2& 1\le i\le \frac{n}{2}+1\\ n-i+1& \frac{n}{2}+1<i\le n\end{array}$$
     整理一下，我们有：
     $$C_i=\{\begin{array}{ll}i-2& 1\le i\le \lfloor \frac{n}{2}\rfloor +1\\ n-i+1& \lfloor \frac{n}{2}\rfloor +1\le i\le n\end{array}$$
     注意到只有 $C_1$ 是负数，考虑列出所有和 $B_1$ 有关的限制条件：
     $$\{\begin{array}{l}{B}_1\le n-1-\sum _{i=2}^n{B}_i\\ B_1\ge \sum _{i=2}^n{C}_i{B}_i\end{array}$$
     在已知 $B_i(2\le i\le n)$ 的情况下，合法的 $B_1$ 的个数为：
     max ⁡ { 0 , n − ∑ i = 2 n ( C i + 1 ) B i } \begin{aligned}\max\{0, n - \sum \limits_{i = 2}^{n}(C_i + 1)B_i\}\end{aligned} max{0,n−i=2∑n​(Ci​+1)Bi​}​

• 设 $f_j$ 表示 $\sum _{i=2}^n(C_i+1)B_i=j$ 的方案数，最后的答案就为 $\sum _{i=0}^{n-1}(n-j)f_j$。

• 显然 $f_j$ 的转移是一个完全背包，时间复杂度 $\mathcal{O}(n^2)$。

Code 2

#include <bits/stdc++.h>

const int N = 1e4 + 5;
int n, mod, ans, c[N], f[N];

inline void add(int &x, int y)
{
	x += y;
	x >= mod ? x -= mod : 0;
}

int main()
{
	scanf("%d%d", &n, &mod);
	int half = n >> 1;
	for (int i = 2; i <= half + 1; ++i)
		c[i] = i - 1;
	for (int i = half + 2; i <= n; ++i)
		c[i] = n - i + 2;

	f[0] = 1;
	for (int i = 2; i <= n; ++i)
		for (int j = c[i]; j < n; ++j)
			add(f[j], f[j - c[i]]);
	for (int j = 0; j < n; ++j)
		ans = (1ll * (n - j) * f[j] + ans) % mod;
	printf("%d\n", ans);
	return 0;
}