P3435 [POI2006]OKR-Periods of Words

Posted 2021-01-05 kafuuchino

P3435 [POI2006]OKR-Periods of Words

题解传送门

kmp

注意：由于题目说只要A满足是2Q的前缀，所以求的不是严格的最大循环子串（20pts）

我们需要求出的是在主串中最小的，既是前缀又是后缀的子串

利用f数组的性质：前缀i的长度为next[i]的前缀和后缀是相等的（然而我之前并不知道qwq）

（对拍10分钟不如手写小数据（大雾））

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
char a[1000003];
int n,f[1000003];
long long ans;
int main(){
    scanf("%d",&n); scanf("%s",a);
    int j;
    for(int i=1;i<n;++i){
        j=f[i];
        while(j&&a[i]!=a[j]) j=f[j];
        f[i+1]= a[i]==a[j] ? j+1:0;
    }
    for(int i=1;i<=n;++i){
        if(!f[i]) continue;
        j=f[i];
        while(f[j]) f[i]=j=f[j];
        ans+=i-j;
    }
    /*for(int i=1;i<=n;++i){
        if(!f[i]) continue; 
        int len=i-f[i];
        ans+=i-(i%len ? i%len:len);
    } //并不能用在这题上
    // 8 uuauuauu 反例
    // 24*/
    printf("%lld",ans);
    return 0;
}