bzoj3620(KMP)

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

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

解题思路：看一下数据发现15s的时间可用暴力，于是就枚举起点，并都做一次KMP，然后根据函数逐一判断结尾。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
using namespace std;
int k,lent,len;
char c[20001]; char s[20001];
int f[16001];
 
void KMP()
 {
    f[0]=-1; f[1]=0;
    for (int i=1;i<=len;++i)
     {
        int j=f[i-1];
        while (j!=-1 && s[j]!=s[i-1])
         {
            j=f[j];
         }
        if (j==-1) f[i]=0;else
         f[i]=j+1;
      }
 }
 
int main()
 {
    scanf("%s",c); scanf("%d",&k); lent=strlen(c);
    int sum=0;
    for (int i=1;i<=lent;++i)
     {
      for (int j=0;j<=lent-i;++j)
      {
      s[j]=c[j+i-1];
 }
len=lent-i+1;
        KMP();
        for (int j=1;j<=len;++j)
         if (j>=2*k+1)
          {
            int now=j;
            while (now!=0 && (f[now]*2>=j || f[now]<k))
             now=f[now];
            if (now!=0) ++sum;
          }
      }
    cout<<sum;
 }