「BZOJ 3620」似乎在梦中见过的样子

传送门

---

problem

给定一个字符串 $S$，求出形如 $ABA$ 的子串数量（$|A|\ge k$，$|B|\ge 1$）。

数据范围：$|S|\le 1.5\times 10^4$，$1\le k\le 100$。

---

solution

可以先做一下「NOI 2014」动物园。

如果我们枚举 $A$ 的左端点 $l$（令 $|S|=r$），那么对于 $(l,r]$ 的位置 $i$，看是否满足：

1. $A$ 是 $[l,i]$ 的公共前后缀。
2. 若 $A$ 串长是 $L$，则 $2L+1\le i-l+1$。
3. $L\ge k$。

诶除了 $3$ 其他的不就是那道题吗，而 $3$ 好像也只需要特判一下求出 $A$ 串长 $\ge k$ 就可以了。

那这道题就做完啦。

时间复杂度 $O(n^2)$，但是能过。

---

code

#include<bits/stdc++.h>
#define N 15005
using namespace std;
char S[N];
int n,k,ans,nxt[N];
int calc(char *S,int l,int res=0){
	for(int i=2,j=0;i<=l;++i){
		while(j&&S[i]!=S[j+1])  j=nxt[j];
		j+=(S[i]==S[j+1]),nxt[i]=j;
	}
	for(int i=2,j=0;i<=l;++i){
		while(j&&S[i]!=S[j+1])  j=nxt[j];
		j+=(S[i]==S[j+1]);
		while((j<<1)>=i)  j=nxt[j];
		if(j>=k)  res++;
	}
	return res;
}
int main(){
	scanf("%s%d",S+1,&k),n=strlen(S+1);
	for(int i=0;i<n;++i)  ans+=calc(S+i,n-i);
	printf("%d\n",ans);
	return 0;
}