BZOJ 3620: 似乎在梦中见过的样子 KMP

3620: 似乎在梦中见过的样子

Time Limit: 15 Sec  Memory Limit: 128 MB
Submit: 834  Solved: 491
[Submit][Status][Discuss]

Description

“Madoka,不要相信 QB！”伴随着 Homura 的失望地喊叫,Madoka 与 QB 签订了契约.

这是 Modoka 的一个噩梦,也同时是上个轮回中所发生的事.为了使这一次 Madoka 不再与 QB签订契约,Homura 决定在刚到学校的第一天就解决 QB.然而,QB 也是有许多替身的(但在第八话中的剧情显示它也有可能是无限重生的),不过,意志坚定的 Homura 是不会放弃的——她决定

消灭所有可能是 QB 的东西.现在，她已感受到附近的状态,并且把它转化为一个长度为 n 的字符串交给了学 OI 的你.

现在你从她的话中知道 , 所有形似于 A+B+A 的字串都是 QB 或它的替身 , 且len(A)>=k,len(B)>=1 （位置不同其他性质相同的子串算不同子串,位置相同但拆分不同的子串算同一子串）,然后你必须尽快告诉 Homura 这个答案——QB 以及它的替身的数量.

Input

第一行一个字符串,第二行一个数 k

Output

仅一行一个数 ans,表示 QB 以及它的替身的数量

Sample Input

【样例输入 1】
aaaaa
1
【样例输入 2】
abcabcabc
2

Sample Output

【样例输出 1】
6

【样例输出 2】
8

HINT

对于 100%的数据：n<=15000 , k<=100，且字符集为所有小写字母

嗯 KMP 嗯 O(n^2) 嗯 过不了 嗯 不会。。。

翻题解。。。。。。为啥O(n^2)能过啊。。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<map>
#include<set>

using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=15100;

int nt[N];
char a[N];
int main()
{
	scanf("%s",a);int n=strlen(a),len=read(),ans=0;
	register int i,j,k,tmp;
	for(j=0;j<n;j++)
	{
		nt[j+1]=j;
		for(i=j+1;i<=n;i++)
		{
			k=nt[i];tmp=k;
			while(tmp!=j&&tmp-j>(i-j-1)>>1)tmp=nt[tmp];
			if(tmp-j>=len&&(i-j)>len<<1)++ans;
			while(k!=j&&a[k]!=a[i])k=nt[k];
			if(a[k]==a[i])nt[i+1]=k+1;
			else nt[i+1]=j;
		}
	}
	print(ans);puts("");return 0;
}
/*
aaaaa
1

6

abcabcabc
2

8
*/