一开始看到数据范围是15000就往后缀数组上想,想了好久没想出来怎么处理同一子串的问题,后来看了看题解才发现O(n^2)就可以过,用KMP类似NOI2014动物园的算法一样。枚举左端点,对于每个右端点处理出以右端点为结尾最大长度使得从左端点开始的前缀等于以右端点结束的后缀,即next数组。然后一直往前跳,直到长度小于子串长度的一半为止。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 20010
using namespace std;
int next[maxn];
char s[maxn];
int n,k,ans;
void KMP(int p)
{
for (int i=1;i<=n;i++) next[i]=p-1;
for (int i=p+1;i<=n;i++)
{
int j=next[i-1];
while (j!=p-1 && s[j+1]!=s[i]) j=next[j];
if (s[j+1]==s[i]) j++;
next[i]=j;
}
int j=next[p];
for (int i=p+1;i<=n;i++)
{
while (j!=p-1 && s[j+1]!=s[i]) j=next[j];
if (s[j+1]==s[i]) j++;
while ((j-p+1)*2>=(i-p+1)) j=next[j];
if (j-p+1>=k) ans++;
}
}
int main()
{
scanf("%s",s+1);
scanf("%d",&k);
n=strlen(s+1);
for (int i=1;i<=n;i++)
KMP(i);
printf("%d\n",ans);
return 0;
}