bzoj 1511: [POI2006]OKR-Periods of Words （kmp+乱搞）

1511: [POI2006]OKR-Periods of Words

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 244   Solved: 138
[ Submit][ Status][ Discuss]

Description

一个串是有限个小写字符的序列,特别的,一个空序列也可以是一个串. 一个串P是串A的前缀, 当且仅当存在串B, 使得 A = PB. 如果 P A 并且 P 不是一个空串,那么我们说 P 是A的一个proper前缀. 定义Q 是A的周期, 当且仅当Q是A的一个proper 前缀并且A是QQ的前缀(不一定要是proper前缀). 比如串 abab 和 ababab 都是串abababa的周期. 串A的最大周期就是它最长的一个周期或者是一个空串(当A没有周期的时候), 比如说, ababab的最大周期是abab. 串abc的最大周期是空串. 给出一个串,求出它所有前缀的最大周期长度之和.

Input

第一行一个整数 k ( 1 k 1 000 000) 表示串的长度. 接下来一行表示给出的串.

Output

输出一个整数表示它所有前缀的最大周期长度之和.

Sample Input

8
babababa

Sample Output

24

HINT

Source

[ Submit][ Status][ Discuss]

题解：kmp+乱搞

这题要求每个前缀的最长循环节的和。给出最长循环节的定义，假设循环节为q,那么要求该前缀是两个q相连的字符串的子串，并且长度大于q。如一个前缀的最长循环节是他本身那么对答案的贡献为0.

这个不难发现最长的循环节一定循环了不超过两次，因为最短循环节是n-末位失配。那么除去末位失配这一段，剩下的保持不变，然后让末位失配这一段求失配不为0时的最短的失配，就是最长循环节多出来的一块。

如果每次都找肯定不行，所以递推 f[i]=(t[i]?f[t[i]]:i)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 1000003
#define LL long long
using namespace std;
int n,t[N],f[N];
LL ans;
char s[N];
void calc()
{
	int j; t[0]=-1;
	for (int i=0;i<n;i++) {
		j=t[i];
		while (j!=-1&&s[i]!=s[j]) j=t[j];
		t[i+1]=++j; 
	}
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d",&n);
	scanf("%s",s);
	calc();
	for (int i=1;i<=n;i++) {
		int j=t[i]; 
		if (t[i])  f[i]=f[t[i]];
		else f[i]=i;
		if (i-t[i]==i)  continue;
		ans+=i-f[i];
	}
	printf("%lld\n",ans);
}