bzoj3620（kmp）

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，且字符集为所有小写字母

---

没想到乳耻暴力
做$n$次$kmp$ $O(n^2)$能过

#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back 
#define pc putchar
#define ll long long
char s[20000];
int k , Next[20000] , cnt[20000];
int read()
{
    int sum = 0;char c = getchar();bool flag = true;
    while( c < '0' || c > '9' ) {if(c == '-') flag = false;c = getchar();}
    while( c >= '0' && c <= '9' ) sum = sum * 10 + c - 48 , c = getchar();
    if(!flag)  sum = -sum;
    return sum;
} 
int main()
{
    scanf("%s",s+1);
    int len = strlen(s+1);
    int j = 0;
    k = read();
    ll ans = 0;
    while(len)
    {
        Next[1] = 0;
        j = 0;
        cnt[1] = 1 >= k ? 1 : 0;
        rep(i,2,len)
        {
            while(j && s[j+1] != s[i]) j = Next[j];
            if(s[j+1] == s[i]) j++;
            Next[i] = j;
            cnt[i] = i >= k ? 1 : 0;
        }
        j = 0;
        rep(i,1,len)
        {
            while(j && s[j+1] != s[i]) j = Next[j];
            if(s[j+1] == s[i]) j++;
            while(( j<<1 ) >= i) j = Next[j];
            ans += cnt[j];
        }
        rep(i,2,len)
            s[i-1] = s[i];
        len--;
    }
    printf("%lld\n",ans);
    return 0;
}