T235569 OKR-Periods of Words

#include<bits/stdc++.h>using namespace std;int k;char st[1000000+100];int f[1000000+100];int main(){

scanf("%d",&k);

scanf("%s",st);

f[0]=-1;

long long ans=0;

for(int i=1;i<k;i++)

{

int j=f[i-1];f[i]=-1;

while(j>=0&&st[j+1]!=st[i])j=f[j];

if(st[j+1]==st[i])f[i]=j+1;

else 

{

f[i]=-1;

continue;

}

}

 

for(int i=1;i<k;i++)

{

if(f[i]==-1)continue;

int j=i;

while(f[j]!=-1)j=f[j];

f[i]=j;

 

ans+=i-j;

 

}

printf("%lld\n",ans);

return 0;

}