bzoj 3620 似乎在夢中見過的樣子
阿新 • • 發佈：2019-01-03
和動物園一樣，維護兩個指標，一個是失配的位置，一個是合法的失配位置。n^2靈異的過了。我wa了好久竟然是把next開成了char
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define md
#define ll long long
#define inf (int) 1e9
#define eps 1e-8
#define N 15010
using namespace std;
char st[N];
int next[N];
ll ans;
int k;
void kmp(char *st)
{
int l=strlen(st+1);
int j=0,now=0;
next[0]=next[1]=0;
for (int i=2;i<=l;i++)
{
while (j&&st[j+1]!=st[i]) j=next[j];
if (st[j+1]==st[i]) j++;
next[i]=j;
while (now&&st[now+1]!=st[i]) now=next[now];
if (st[now+1]==st[i]) now++;
while (2*now>=i) now=next[now];
if (now>=k) ans++;
}
}
int main()
{
scanf("%s%d",st+2,&k);
ans=0;
int n=strlen(st+2);
for (int i=1;i<=n;i++) kmp(st+i);
printf("%lld\n",ans);
return 0;
}