bzoj 3620 似乎在梦中见过的样子（KMP）

 1 #include<cstdio>
 2 #include<cstring>
 3 using namespace std;
 4 
 5 typedef long long ll;
 6 const int N = 2e4+10;
 7 
 8 int n,K;
 9 ll ans;
10 
11 int f[N];
12 char s[N];
13 
14 void KMP(char *s) {
15     int n=strlen(s+1);
16     int j=0,now=0;
17     for(int i=2;i<=n;i++) {
18         while(j&&s[j+1]!=s[i]) j=f[j];
19         if(s[j+1]==s[i]) j++;
20         f[i]=j;
21         
22         while(now&&s[now+1]!=s[i]) now=f[now];
23         if(s[now+1]==s[i]) now++;
24         while(2*now>=i) now=f[now];
25         if(now>=K) ans++;  
26     }
27 }
28 int main()
29 {
30     scanf("%s%d",s+2,&K);
31     int n=strlen(s+2);
32     for(int i=1;i<=n;i++) KMP(s+i);
33     printf("%lld\n",ans);
34     return 0;
35 }