1. 程式人生 > >BZOJ 1511 [POI2006]OKR-Periods of Words KMP

BZOJ 1511 [POI2006]OKR-Periods of Words KMP

題意:
求一個串的所有字首的最長週期長度之和,特別的,週期為自己的串的最長週期長度視作0.
解析:
直接求一下next,之後把所有的next向前找到最後一個非零地方的Next。
然後掃一遍對於每個next非零位置的週期來說就是i-new_next[i]
還是之前的那個性質,n-next[i]是最小迴圈週期,推一下就變成最長了。
程式碼:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 1000100
using namespace std;
typedef long long ll;
int
n; int ne[N]; char s[N]; void getnext() { ne[0]=-1; int i=0,j=-1; while(i<n) { if(j==-1||s[i]==s[j]) { i++,j++; ne[i]=j; }else j=ne[j]; } } int main() { #ifndef ONLINE_JUDGE freopen("1511.in","r",stdin); freopen("1511.out"
,"w",stdout); #endif scanf("%d",&n); scanf("%s",s); getnext(); ll ans=0; for(int i=1;i<=n;i++) { if(!ne[i])continue; while(ne[ne[i]]) ne[i]=ne[ne[i]]; } for(int i=1;i<=n;i++) { if(ne[i]!=0) ans+=i-ne[i]; } printf
("%lld\n",ans); #ifndef ONLINE_JUDGE fclose(stdin);fclose(stdout); #endif }