有些类似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; }