《ybtoj高效进阶》第二部分第三章例题4 子串拆分

题目大意

问你在一个字符串中，有多少个$ABA$形的子串（B的长度>=1）。
位置不同其他性质相同的子串算不同子串，位置相同但拆分不同的子串算同一子串。

思路

我们暴力枚举该字符串左端点，然后对新串KMP求解。
大概就是考虑现在匹配到的位置前面是否有同样的一段且中间至少空了一个。
code：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
using namespace std;
long long p[1000006],n,m,ans,k;
string b;
inline void f()
{
	p[1]=0;
	long long i=1,j=0;
	while (i<m)
	{
		while (j&&b[1+j]!=b[1+i]) j=p[j];
		if (b[1+j]==b[1+i]) j++;
		p[i+1]=j;
		i++;
	}
	return;
}
inline void kmp()
{
	long long i=1,j=0;
	while (i<m)
	{
		while (j&&b[1+j]!=b[1+i]) j=p[j];
		if (b[1+j]==b[1+i]) j++;
		while (j*2>=i+1) j=p[j];
		if (j>=k) ans++;
		i++;
	}
	return;
}
long long find(long long x)
{
	if (p[x]!=0) return p[x]=find(p[x]);
	return x;
}
char y;
int main()
{
	cin>>b>>k;
	m=b.size();
	b.insert(0," ");
	for (int i=0;i<m-k*2;i++) f(),kmp(),b.erase(1,1);
	cout<<ans;
    return 0;
}