CF432D Prefixes and Suffixes(KMP)

文章目录

【注意】最后更新于 5年前,文中内容可能已过时,请谨慎使用。

题目链接

洛谷

CF

题意简述

给你一个字符串,分别求出每个长度的既是其前缀又是其后缀的串出现的次数。

字符串长度小于等于 105

简要做法

第一眼,SAM 裸题。

然后意识到只用管前缀,KMP 可以达到同样的效果。

做法和 SAM 完全一样:从末尾跳 next 来找到所有前后缀相同的位置(也就是 SAM 的接受状态),然后倒序枚举 fnext[i]+=fi(也就是 SAM 的统计子树和)。

多说两句:fail、parent 和 next 其实是一样的东西,但 KMP / AC 自动机的状态是所有前缀,SAM 的状态是所有子串(并被压缩为了 right 集合等价类)。

参考代码

复制
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int N = 100010;
char s[N];
int n, nxt[N], stk[N], top, f[N];
int main()
{
int i, k;
scanf("%s", s + 1);
n = strlen(s + 1);
for (i = 2, k = 0; i <= n; ++i)
{
while (k && s[i] != s[k + 1]) k = nxt[k];
if (s[i] == s[k + 1]) ++k;
nxt[i] = k;
}
for (i = n; i >= 1; --i)
{
++f[i];
f[nxt[i]] += f[i];
}
for (i = n; i; i = nxt[i]) stk[++top] = i;
printf("%d\n", top);
while (top)
{
printf("%d %d\n", stk[top], f[stk[top]]);
--top;
}
return 0;
}

评论正在加载中...如果评论较长时间无法加载,你可以 搜索对应的 issue 或者 新建一个 issue