问题
给定一个字符串
暴力做法
考虑动态规划,记
由于一个字符串最多会有
显然不能接受,为了加速转移,首先给出一些引理。
引理与证明
知乎传公式太麻烦了,证明过程留作读者自行思考(误,可以自行翻阅参考资料或前往我的博客阅读 https:// xlor.cn/2019/11/mpf )
定义
记字符串
周期:若
border:若
周期和 border 的关系
证明:
若
若
引理 1
下图中,相同颜色的位置表示字符对应相同。
![e0cae78b40250ea97a6f07820bfa2620.png](https://i-blog.csdnimg.cn/blog_migrate/82abb4f32fb21ce2448fd063eaa6fc78.jpeg)
引理 2
引理 3
引理 4
-
;
- 如果
,那么
;
- 如果
,那么
。
推论
做法
有了上述结论后,我们现在可以考虑如何优化 dp 的转移。
回文树上的每个节点
根据上面证明的结论,如果使用 slink 指针向上跳的话,每向后填加一个字符,只需要向上跳
下面我们考虑如何更新 g 数组和 dp 数组。以下图为例,假设当前枚举到第
![9955d9c6d1dc6055941d56c66da66879.png](https://i-blog.csdnimg.cn/blog_migrate/a60579833e26d1b09f9cdf7533fb019b.jpeg)
最后,上述做法的正确性依赖于:如果
Codeforces 932G Palindrome Partition
构造字符串
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
const int maxn = 1000000 + 5;
inline int add(int x, int y) {
x += y;
return x >= mod ? x -= mod : x;
}
namespace pam {
int sz, tot, last;
int ch[maxn][26], len[maxn], fail[maxn];
int cnt[maxn], dep[maxn], dif[maxn], slink[maxn];
char s[maxn];
int node(int l) {
sz++;
memset(ch[sz], 0, sizeof(ch[sz]));
len[sz] = l;
fail[sz] = 0;
cnt[sz] = 0;
dep[sz] = 0;
return sz;
}
void clear() {
sz = -1; last = 0;
s[tot = 0] = '$';
node(0); node(-1);
fail[0] = 1;
}
int getfail(int x) {
while (s[tot - len[x] - 1] != s[tot]) x = fail[x];
return x;
}
void insert(char c) {
s[++tot] = c;
int now = getfail(last);
if (!ch[now][c - 'a']) {
int x = node(len[now] + 2);
fail[x] = ch[getfail(fail[now])][c - 'a'];
dep[x] = dep[fail[x]] + 1;
ch[now][c - 'a'] = x;
dif[x] = len[x] - len[fail[x]];
if (dif[x] == dif[fail[x]]) slink[x] = slink[fail[x]];
else slink[x] = fail[x];
}
last = ch[now][c - 'a'];
cnt[last]++;
}
}
using pam::len;
using pam::fail;
using pam::slink;
using pam::dif;
int n, dp[maxn], g[maxn]; char s[maxn], t[maxn];
int main() {
pam::clear();
scanf("%s", s + 1);
n = strlen(s + 1);
for (int i = 1, j = 0; i <= n; i++) t[++j] = s[i], t[++j] = s[n - i + 1];
dp[0] = 1;
for (int i = 1; i <= n; i++) {
pam::insert(t[i]);
for (int x = pam::last; x > 1; x = slink[x]) {
g[x] = dp[i - len[slink[x]] - dif[x]];
if (dif[x] == dif[fail[x]]) g[x] = add(g[x], g[fail[x]]);
if (i % 2 == 0) dp[i] = add(dp[i], g[x]);
}
}
printf("%d", dp[n]);
return 0;
}
参考资料
- EERTREE: An Efficient Data Structure for Processing Palindromes in Strings
- Palindromic tree
- 2017 年 IOI 国家候选队论文集 回文树及其应用 翁文涛
- 2019 年 IOI 国家候选队论文集 子串周期查询问题的相关算法及其应用 陈孙立
- 字符串算法选讲 金策
- A bit more about palindromes
- A Subquadratic Algorithm for Minimum Palindromic Factorization