AtCoder Beginner Contest 249 - E - RLE（前缀和优化dp）

皮皮带王

已于 2022-04-27 18:54:39 修改

阅读量567

点赞数 6

分类专栏：动态规划文章标签：算法

于 2022-04-27 14:03:34 首次发布

本文链接：https://blog.csdn.net/m0_51979134/article/details/124447096

版权

动态规划专栏收录该内容

3 篇文章 1 订阅

订阅专栏

题目链接：E - RLE (atcoder.jp)

题目大意：给定一个字符串（由小写字母组成）转换方法如下：

$a$ -> $a1$

$aaa$ -> $a3$

$aaabbcccc$ -> $a3b2c4$

$aaaaaaaaaa$ -> $a10$

转换方法大概就是这样，详细的文字说明就八写了。总之就是把连续的字母转化成字母+字母个数的形式。注意 $a9$ 和 $a10$ 的长度是不一样的。现要统计出所有长度为 $n$ 的字符串中转换之后的长度严格小于 $n$ 的数量。

题解：考虑使用 $dp$ 求解。 $f[i][j]$ ：原串长度为 $i$ ，转换后长度为 $j$ 的数量。

我们可以考虑在一个字符串后面接一个长度为 $k$ 的串，接上的串所有字符相同且与原串最后一个字符不同。递推式如下：

$f[i][j]=\sum\limits_{k=1}^{i}f[i-k][j-get(k)]\times 25$

$get(k)$ 含义：考虑接上的串的长度 $k$ ，因为所有字符相同，发现该串的生成串长度只有 $4$ 种可能：

$k\in [1, 10) ,get(k) = 2$

$k\in [10, 100) ,get(k) = 3$

$k\in [100, 1000) ,get(k) = 4$

$k\in [1000, 10000) ,get(k) = 5$

直接实现代码，时间复杂度为 $\mathcal{O}(n^3)$

for (int i = 1; i <= n; i++)
		for (int j = 1; j <= 2*i; j++) //生成串最长可能为原串两倍
            for (int k = 1; k <= i && get(k) <= j; k++) 
			 	f[i][j] += f[i-k][j - get(k)] * 25; //接上的串字符于之前的串最后一个字符不同，即25种可能性

但是本体数据范围是 $1\leq n\leq 3000$ ，所以直接暴力显然是无法通过本题的，接下来考虑优化。

我们观察到

可以把 $k\in[1, i]$ 分成许多个区间，即 $[1, 9], [10,99],[100, 999]...$ ，对于同一区间里的 $k$ ，所对应的 $get(k)$ 是一样的，所以得到：

$f[i][j] = (\sum\limits_{k=1}^{9} f[i-k][j-2]+ \sum\limits_{k=10}^{99} f[i-k][j-3]+ ... + \sum\limits_{k=got(p)}^{i} f[i-k][j-p])\times25$

$got()$ 类似于 $get()$ 的反向操作，即输入生成串的长度，返回原串的最短长度。

现在我们只需要用前缀和数组 $pref[i][j]$ 来维护前缀信息，即

$pref[i][j] = \sum\limits_{k=1}^{k=i}f[k][j]$

这样就可以把 $\sum\limits_{k=1}^{9} f[i-k][j-2]$ 这样的 $j$ 相同的区间和转化如下：

$\sum\limits_{k=1}^{9} f[i-k][j-2] = pref[i-1][j-2]-pref[i-9-1][j-2]$

因为最多只会有 $4$ 个 $j$ 不同的区间和，所以时间复杂度为 $\mathcal{O}(4\times n^2)$

完整代码如下：

#include <bits/stdc++.h>

using namespace std;

#define int long long
const int N = 3010;

int f[N][2*N], n, p, pref[N][2*N];
  
int get(int x) {
	if(1 <= x && x < 10) return 2;
	else if(x >= 10 && x <= 99) return 3;
	else if(x >= 100 && x <= 999) return 4;
	else if(x >= 1000 && x <= 9999) return 5;
}  

int got(int x) {
	if(x == 2) return 9;
	if(x == 3) return 99;
	if(x == 4) return 999;
	if(x == 5) return 9999;
}

signed main() {
	cin >> n >> p;
	for(int i=1; i<=n; i++) {
		f[i][get(i)] = 26; 
		pref[i][get(i)] = pref[i-1][get(i)] + 26;
	}

	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= 2*i; j++) {
			int k, S = i-1, t2 = 0, SS = i;
			for (k = 2; k <= j && got(k) < SS; k++) {
				f[i][j] = ((((pref[S][j-k] - pref[SS-got(k)-1][j-k]) + p) % p * 25) % p  + f[i][j]) % p; 
 				S = SS - got(k)-1;
			}
			f[i][j] = ((((pref[S][j-k] - pref[0][j-k]) + p) % p * 25) % p + f[i][j]) % p;
			pref[i][j] = (pref[i-1][j] + f[i][j]) % p;
		}
	int res = 0;
	for (int j = 1; j < n; j++)
		res = (res + f[n][j]) % p;

	cout << res  << '\n';
}