P3435 [POI2006]OKR-Periods of Words（kmp）

#include<cstdio> #include<iostream> using namespace std; const int N=1000005; int n; long long ans; char s[N]; int nxt[N]; int main(){ scanf("%d",&n); scanf("%s",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++){ j=i; while(nxt[j]) j=nxt[j]; if(nxt[i]!=0) nxt[i]=j; ans+=i-j; } printf("%lld",ans); return 0; }