poj 2406(KMP)

poj 2406
可以知道,在一个字符串里的最短周期是$ms=n-f[n]​$, 其中$f​$是kmp中的失配函数。
如果$ms|n$, 那么输出$n/ms$, 否则输出$1$.

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
#include<cstdio>
#include<algorithm>
#include<cstring>
#define ms(i,j) memset(i,j, sizeof i);
using namespace std;
const int MAXN = 1000000 + 5;
int n;
char s[MAXN];
int f[MAXN];
int main()
{
while (scanf("%s", s)&&(s[0]!='.'))
{
n = strlen(s);
f[0] = f[1] = 0;
for (int i=1;i<n;i++)
{
int j = f[i];
while (j && s[i]!=s[j]) j = f[j];
f[i+1] = (s[i]==s[j]) ? (j+1) : (0);
}
int ans = n-f[n];
if (n%ans==0) ans = n/ans; else ans = 1;
printf("%d\n", ans);
}
return 0;
}

------ 本文结束 ------