【bzoj3620】似乎在梦中见过的样子

有些类似noi2014的动物园，也是对于KMP算法的一个应用，思想就是枚举前缀然后预先留出k的位置，对于自身KMP，当next数组>k时就计入答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>

using namespace std;
#define N 15005
char s[N];
int f[N];
int k,l,ans,lim;
int main()
{
    scanf("%s%d",s+1,&k);
    l=strlen(s+1);lim=l-k*2;
    for (int p=0;p<lim;p++)//枚举左端点，对每一个左端点做KMP 
    {
        for (int j=0,i=2;i+p<l;i++)//处理next（f）数组 
        {
            while (j&&s[j+p+1]!=s[i+p])j=f[j];
            if (s[i+p]==s[j+p+1])j++;f[i]=j; 
        }
        for (int j=0,i=k+1;i+p<=l;i++)//类似noi2014的动物园 
        {
            while (j&&s[i+p]!=s[j+p+1])j=f[j];
            if (s[i+p]==s[j+p+1])j++;
            while ((j<<1)>=i)j=f[j];if (j>=k)ans++;//当前缀与后缀都>=k即j>=k时并且<=i>>1时计入答案 
        }
    }
    cout<<ans;
    return 0;
}