BZOJ 1511 [POI2006]OKR-Periods of Words KMP
阿新 • • 發佈:2018-12-24
題意:
求一個串的所有字首的最長週期長度之和,特別的,週期為自己的串的最長週期長度視作0.
解析:
直接求一下next,之後把所有的next向前找到最後一個非零地方的Next。
然後掃一遍對於每個next非零位置的週期來說就是i-new_next[i]
還是之前的那個性質,n-next[i]是最小迴圈週期,推一下就變成最長了。
程式碼:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1000100
using namespace std;
typedef long long ll;
int n;
int ne[N];
char s[N];
void getnext()
{
ne[0]=-1;
int i=0,j=-1;
while(i<n)
{
if(j==-1||s[i]==s[j])
{
i++,j++;
ne[i]=j;
}else j=ne[j];
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("1511.in","r",stdin);
freopen("1511.out" ,"w",stdout);
#endif
scanf("%d",&n);
scanf("%s",s);
getnext();
ll ans=0;
for(int i=1;i<=n;i++)
{
if(!ne[i])continue;
while(ne[ne[i]])
ne[i]=ne[ne[i]];
}
for(int i=1;i<=n;i++)
{
if(ne[i]!=0)
ans+=i-ne[i];
}
printf ("%lld\n",ans);
#ifndef ONLINE_JUDGE
fclose(stdin);fclose(stdout);
#endif
}