..这题其实挺好想的,然而我想到正解以后,觉得不能过,然后摒弃了,打了个SA调了半天才发现位置不同才算不同否则全部相同。。
而且我正解时间复杂度分析错了,,以为是N^2结果实现以后才发现随便搞一搞就能线性。。
正解就是kmp暴力,,枚举左端点然后直接kmp找出所有的A,然后看看B和A符不符合条件就行了。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
int n,m;
const int N=1e5+5;
char s[N];
int next[N],g[N];
int ans;
int main()
{
scanf("%s%d",s+1,&m);
n=strlen(s+1);
int len=n-m*2;
int k;
fo(i,1,len)
{
next[i]=k=i-1;
g[i]=i;
fo(j,i+1,n)
{
while (k!=i-1&&s[j]!=s[k+1])k=next[k];
if (s[j]==s[k+1])k++;
next[j]=k;
g[j]=(k<i+m-1)?j:g[k];
if (g[j]<((i+j)>>1))ans++;
}
}
printf("%d\n",ans);
return 0;
}