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

时间:2022-01-10 22:26:25

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还是不熟。。

http://blog.csdn.net/cymxyym/article/details/37569675

/*我们可以做n次kmp: 第i次,以[i,n]为模板 构造出next数组,
然 后 对 于 [i,j-1] 这 个 子 串 , 我 们 知 道 [i,next[j]] 与
[j-1-(next[j]-i+1),j-1]是相同的,并且这个已经是使 A 串长度最
大化的了。 
如果两个 A 串不重叠,且长度大于 k 就++ans,否则继续检查
next[next[j]] 
如果就这样的话,是会达到n^3的复杂度的(比如样例 1) ,
 记一下大于等于k的最小是多少就行了,假设这
个记为 L[i] , 然 后 如 果 next[i]-i【初始位置】>=k , 就
L[i]=min{next[i],L[next[i]]},否则 L[i]=L[next[i]] 这样就可
以做到n^2了。 */

果然还是太弱。。。

#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
char s[150001];
int next[160001];
int L[150001];
int main()
{
     scanf("%s",s+1);
     int len=strlen(s+1);
     int i,j;
     int k;
     scanf("%d",&k);
     int ans=0;
     for(i=1;i<=len-k*2;i++)//枚举开头 
     {
          next[i]=i;
		  next[i+1]=i;
          L[i]=2147483647;
		  L[i+1]=2147483647;
          for(j=i+1;j<=len;j++)
          {
               int t=next[j];  
               while(t>i&&s[t]!=s[j])
			        t=next[t]; 
               if(s[t]!=s[j])
			        next[j+1]=i;  
               else
	                next[j+1]=t+1; //求next数组 
               if(next[j+1]-i>=k)
                    L[j+1]=min(next[j+1],L[next[j+1]]);
               else
	                L[j+1]=L[next[j+1]];
               if(L[j+1]-i>=k&&j-(L[j+1]-i)>=L[j+1])
			        ans++;
          }
     }
     printf("%d\n",ans);
     return 0;
}