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

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

Time Limit: 15 Sec   Memory Limit: 128 MB
Submit: 1202   Solved: 697
[ 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，且字符集为所有小写字母

Source

2014湖北省队互测week2

[ Submit][ Status][ Discuss]



这一题看起来很难，于是先打了个暴力，谁知交上去竟过了？

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 15005;
char s[N];
int n,ans,next[N],k;
void init(){
	memset(next,0,sizeof(next));
}
void get(int x){
	init();
	next[x]=x-1;
	int j=x-1;
	for(int i=x;i<n;i++){
		while(s[i+1]!=s[j+1]&&j>x-1) j=next[j];
		if(s[i+1]==s[j+1]) j++;
		next[i+1]=j;
	}
	j=x-1;
	for(int i=x;i<=n;i++){
		while(j>x-1&&s[j+1]!=s[i+1]) j=next[j];
		if(s[j+1]==s[i+1]) j++;
		while((j-x+1)*2>=(i+1-x+1)) j=next[j];
		if(j-x+1>=k) ans++;
	}
}
int main(){
	scanf("%s",s+1);
	n=strlen(s+1);
	scanf("%d",&k);
	for(int i=1;i<=n;i++) get(i);
	printf("%d",ans);
	return 0;
}