BZOJ 3620 似乎在夢中見過的樣子 KMP+暴力

BZOJ 3620 似乎在夢中見過的樣子 KMP+暴力

題目大意：給定一個字符串，求這個字符串有多少個子串滿足這個子串可以拆分成ABA的形式，其中|A|>=k,|B|>=1

夢の中で逢った、ような…...

n<=15000 顯然是直接給你暴力用的- -

枚舉子串的左端點，然後枚舉右端點

對於每個子串S我們要判定是否存在一個長度在[k,|S|-1>>1]之間的前綴與後綴匹配

那我們就求出長度不超過|S|-1>>1的最長前後綴，判斷是否>=k即可

這怎麼和去年NOI的某道【植物園】很像。。。呵呵。。。

於是常數略大，光榮墊底

#include 
#include 
#include 
#include 
#define M 15010
using namespace std;
int n,k,ans;
int next[M];
char s[M];
void KMP(char *s)
{
	int i,fix=0;
	for(i=2;s[i];i++)
	{
		while( fix && s[fix+1]!=s[i] )
			fix=next[fix];
		if( s[fix+1]==s[i] )
			++fix;
		next[i]=fix;
	}
	fix=0;
	for(i=1;s[i];i++)
	{
		while( fix && s[fix+1]!=s[i] )
			fix=next[fix];
		if( s[fix+1]==s[i] )
			++fix;
		while( fix<<1>=i )
			fix=next[fix];
		if(fix>=k)
			++ans;
	}
}
int main()
{
	int i;
	scanf("%s",s+1);
	cin>>k;
	for(i=0;s[i+1];i++)
		KMP(s+i);
	cout<