BZOJ 1511: [POI2006]OKR-Periods of Words

題目描述

/*
問題即求某一前綴A=XYZX的最小X的長度，可根據KMP中的nxt求出 
但直接作會超時 
考慮已經求過的某一串A是另外一個串B的前綴，那麼在求B的對答案的貢獻時會重複一次求A的過程 
因此能夠在求完A以後更新A的nxt爲求出X的位置，這樣求B時就會直接跳到X的位置 
*/
#include<complex>
#include<cstdio>
using namespace std;
const int N=1e6+7;
int n;
long long ans;
int nxt[N];
char s[N];
int main()
{
    scanf("%d%s",&n,s+1);
    int j=0;
    for(int i=2;i<=n;i++)
    {
        while(j && s[i]!=s[j+1])j=nxt[j];
        if(s[i]==s[j+1])j++;
        nxt[i]=j;
    }
    for(int i=1;i<=n;i++)
    {
        int j=i;
        while(nxt[j])j=nxt[j];
        if(nxt[i])nxt[i]=j;
        ans+=i-j;
    }
    printf("%lld\n",ans);
    return 0;
}