1511: [POI2006]OKR-Periods of Words
Time Limit: 5 Sec Memory Limit: 64 MBSubmit: 290 Solved: 170
[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
babababa
Sample Output
24
(1)
题解:kmp+乱搞
这题要求每个前缀的最长循环节的和。给出最长循环节的定义,假设循环节为q,那么要求该前缀是两个q相连的字符串的子串,并且长度大于q。如一个前缀的最长循环节是他本身那么对答案的贡献为0.
这个不难发现最长的循环节一定循环了不超过两次,因为最短循环节是n-末位失配。那么除去末位失配这一段,剩下的保持不变,然后让末位失配这一段求失配不为0时的最短的失配,就是最长循环节多出来的一块。
如果每次都找肯定不行,所以递推 f[i]=(t[i]?f[t[i]]:i)
直接求一下next,之后把所有的next向前找到最后一个非零地方的Next。
然后扫一遍对于每个next非零位置的周期来说就是i-new_next[i]
还是之前的那个性质,n-next[i]是最小循环周期,推一下就变成最长了。
for(int i=1;i<=n;i++)
{
if(!next[i])continue;
while(next[next[i]])
next[i]=next[next[i]];
}
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return f*x;
}
const int N=1000100;
int n,nxt[N],f[N];
char s[N];
int main()
{
n=read();ll ans=0;
scanf("%s",s);
for(int i=1,j;i<n;i++)
{
j=nxt[i];
while(j&&s[i]!=s[j])j=nxt[j];
if(s[j]==s[i])nxt[i+1]=j+1;
else nxt[i+1]=0;
}
for(int i=1,j;i<=n;i++)
{
j=nxt[i];
if(j)f[i]=f[j];
else f[i]=i;
if(i-j==i)continue;
ans+=i-f[i];
}
printf("%lld\n",ans);
return 0;
}
/*
8
babababa
*/