【AtCoder Beginner Contest 253】E - Distance Sequence

传送门

题意

给定 $N,M,K$ ，数组 $a_1,a_2,...,a_N$，满足如下两个条件

$$\begin{array}{rlrl}& 1<=A_i<=M，& & (1<=i<=N)\\ & |A_i-A_{i+1}|>=K，& & (1<=i<=N-1)\end{array}$$

求构成这样的序列的总方案数。

---

思路

$$\begin{array}{rl}定义：& \\ & \\ & dp[i][j]：表示确定A的前i项的方法数，使得第i项为j\\ & \\ 转移方程：& \\ & \\ & dp[i+1][j]=(dp[i][1]+...+dp[i][j-k])+(dp[i][j+k]+...+dp[i][M])\end{array}$$

显然如上转移，时间复杂度为 $O(N{M}^2)$，不可接受。

那么我们可以在计算 $dp[i+1]$ 时，先处理好 $dp[i]$ 的状态，然后进行 $O(1)$ 得到。

那么时间复杂度很好的优化到了 $O(NM)$。

解法中会用到：

• 二维压一维

  • 计算当前 $i+1$ 的状态，只与 $i$ 层状态有关。

• 前缀和优化

  • 前缀和数组 $s$
  • $dp[i]=s[m]-s[j+k-1]+s[j-k+1]$

还需注意的是当 $k==0$ 时：

• 无需后两项，$dp[i]=s[m]$

---

代码

• 前缀和 + 动态规划

int n, m, k;
int f[M], s[M];

void solve() {
	cin >> n >> m >> k;
	for (int i = 1; i <= m; i++) f[i] = 1; // i == 1;
	for (int i = 2; i <= n; i++) { // i > 1
		s[0] = 0;
		for (int j = 1; j <= m; j++)
			s[j] = (s[j - 1] + f[j]) % mod;
		int sum = s[m];
		for (int j = 1; j <= m; j++) {
			int l = max(1ll, j - k + 1);
			int r = min(m, j + k - 1);
			if(k > 0) f[j] = (sum - s[r] + s[l - 1] + mod) % mod;
			else f[j] = sum % mod;
 		}
	}
	int res = 0;
	for (int i = 1; i <= m; i++) res = (res + f[i]) % mod;
	cout << res << endl;
}