【bzoj3620】似乎在夢中見過的樣子
阿新 • • 發佈:2019-01-02
有些類似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;
}