bzoj 3620: 似乎在梦中见过的样子 kmp算法+暴力

题意

“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 以及它的替身的数量.
n<=15000 , k<=100，且字符集为所有小写字母

分析

一开始就往sa和sam上面去想了，发现还是不会做，后来看了一波题解，发现只要 n2 就能过。

对于一个区间[l,r]，如果有一个w满足长为w的前缀等于长为w的后缀且w>=k,w<=(r-l)/2那么该区间即为合法区间。
那么我们可以枚举左端点l，然后求出其next数组，用一个辅助数组f[i]表示i不停地往next走，能走到的最小的且不小于k的长度是多少，然后扫一遍统计右端点贡献即可。

代码

#include
#include
#include
#include
#include
#define N 15005
using namespace std;

int k,f[N],n,next[N];
char s[N];

void get_next(int x)
{
    next[x]=x-1;next[x-1]=x-2;
    int i=x+1,j=x-1;
    while (i<=n)
        if (s[j+1]==s[i]||j==x-2)
        {
            j++;
            next[i]=j;
            i++;
        }
        else j=next[j];
}

int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    scanf("%d",&k);
    int ans=0;
    for (int i=1;i<=n-k*2;i++)
    {
        get_next(i);
        for (int j=i+1;j<=n;j++)
        {
            f[j]=0;
            if (next[j]-i+1>=k) f[j]=next[j]-i+1;
            if (f[next[j]]>=k) f[j]=min(f[j],f[next[j]]);
            if (f[j]>=k&&f[j]<=(j-i)/2) ans++;
        }
    }
    printf("%d",ans);
    return 0;
}