洛谷傳送門，BZOJ傳送門

OKR-Period of Words

Description

Input

Output

Sample Input

Sample Output

　　分析：

　　又是一道需要深入理解$next[]$數組的題。

　　因為$next[i]$表示的是第$i-1$個前綴的最長公共前後綴，那麽不難想到，在$j<=i$且$next[j]$不為$0$的情況下可以得到一個最短後綴，用原串長度減去即可得到答案。

　　Code：

//It is made by HolseLee on 11th Aug 2018
 
//Luogu.org P3435
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<iomanip>
#include<algorithm>
using namespace std;

const int N=2e6+7;
int n,nxt[N],k,cnt;
long long ans;
char s[N];

int main()
{
    scanf(
 
"%d%s",&n,s);
    nxt[0]=nxt[1]=0;k=0;
    for(int i=1;i<n;++i){
        while(k&&s[i]!=s[k])
        k=nxt[k];
        nxt[i+1]=(s[i]==s[k]?++k:0);
    }
    int ka;
    for(int i=1;i<=n;++i){
        ka=i;
        while(nxt[ka])ka=nxt[ka];
        if(nxt[i])nxt[i]=ka;
        ans+=(i-ka);
    }
    printf("%lld ",ans);
    return 0;
}

洛谷P3435 [POI2006]OKR-Period of Words [KMP]