【CF】1989E-Distance to Different 题解

传送门：1989E
标签：动态规划

题目大意

考虑一个由n个整数组成的数组a，其中每个元素都在1到k之间，并且每一个从1到k的整数至少出现一次。让b数组通过以下方式构造：对于a中的第i个元素，b[i]是到与a[i]不相等的最近元素的距离。换句话说，bi=minj∈[1,n],aj≠ai∣i−j∣。例如，如果a=[1,1,2,3,3,3,3]，那么b=[2,1,1,1,2,1,1]。计算所有可能的a数组中可以得到的不同b数组的数量，并对998244353取模后输出。

输入：输入只有一行，包含两个整数n和k（2≤n≤2⋅10⁵；2≤k≤min(n,10)）。

输出：输出一个整数——可以得到的不同b数组的数量，对998244353取模后输出。

算法分析

• 考虑 $a$ 中的一个等元素块。如果我们把 $a$ 分成这样的块，我们可以单独考虑每个块 — 对于每个元素，我们只需要距离最近边界的距离（除了第一个和最后一个块，我们只考虑一个边界）。很容易看出，如果块的长度是偶数（设为 $2x$），则这些块内的距离为 [1, 2, 3, …, x-1, x, x, x-1, …, 3, 2, 1]。如果块的长度是奇数（设为 $2x-1$），则距离为 [1, 2, 3, …, x-1, x, x-1, …, 3, 2, 1]。现在我们不需要 $a$ 中的实际值，我们只需要关于等元素块的信息。
• 我们需要至少有 $k$ 个等元素块，因为如果块的数量小于 $k$，数组 $a$ 不可能有 $k$ 个不同的值（如果有至少 $k$ 个块，就可以为每个块分配一个值，使得从 1 到 $k$ 的每个整数都出现）。因此，我们需要计算分割数组为至少 $k$ 块的方法数。
• 然而，只有当数组分割与结果数组 $b$ 之间存在双射关系时，这种方法才有效。不幸的是，有些数组分割可能会导致相同的结果数组：例如，距离数组 [1, 1, 1] 可以通过将数组分成四个大小为 1 的块得到，也可以通过将数组分成三个块，其中中间的块大小为 2。所以，如果有一个大小为 2 的块，并且不是分割的第一个或最后一个块，它可以替换为两个大小为 1 的块，除了块的数量增加外，其他都不会改变。
• 这是唯一需要处理的情况：任何大于 2 的块都可以由其内部的值唯一确定（如果有一个值 $x$ 大于它的邻居，那么它是大小为 $2x-1$ 的块的中心；如果有一对值 $x$ 大于左边和右边的值，那么它是大小为 $2x$ 的块的中心）。因此，我们需要计算将大小为 $n$ 的数组分割为至少 $k$ 个块的方式数，以便只有第一个和最后一个块可以具有大小为 2。这可以通过以下动态规划完成：令 $d{p}_{t,j}$ 是将前 $i$ 个元素分割为 $j$ 个块的方法数。
• 这看起来像是 $O(n^3)$，但我们可以通过两种改进来加速：如果 $j>k$，则它们与 $j=k$ 相似，所以我们可以限制 $j$ 的值为 $k$ 并说 $d{p}_{t,j}$ 是将前 $t$ 个元素分割为至少 $j$ 个块的方法数；我们可以用部分和代替 $O(n)$ 进行转移。结合这两种优化，我们得到了一个解决方案，时间复杂性为 $O(nk)$。

代码实现

#include<iostream>
using namespace std;
#include <algorithm>
#include <cstring>
#include <vector>
#include <unordered_map>
#include <map>
#define int long long
const int N = 2e5+10,INF = 1e18,M = N+N,mod = 998244353;
int f[N][11];
int g[11];
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,k;
	cin>>n>>k;
	f[0][0] = 1;
	g[0] = 1;
	for(int i = 1;i<=n;i++)
	{
		for(int j = k;j>=1;j--)
		{
			f[i][j] = g[j-1];
			if(j==k) f[i][j] = (f[i][j] + g[j])%mod;
			if(i>2&&i!=n)
			{
				f[i][j] = (f[i][j] - f[i-2][j-1] + mod)%mod;
				if(j==k) f[i][j] = (f[i][j] - f[i-2][j] + mod)%mod;
			}
			g[j] = (g[j] + f[i][j])%mod;
		}
	}
	cout<<f[n][k]<<'\n';
	return 0;
}