[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^2能过2333…
原本怒想一手nlogn发现刚不动然后才知道N^2可过。。
你就枚举每个起始点往后跑一遍KMP，记录一个go[i]表示这个点能满足>=K且长度最小的长度，如果不存在就INF
然后扫一遍就行了。。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
int p[16010],go[16010],K,ans,tmp;
char ch[16010];
int main()
{
    scanf("%s",ch+1);int n=strlen(ch+1);
    scanf("%d",&K);tmp=0;
    go[0]=999999999;
    for(int i=1;i<=n;i++)
    {
        p[i]=0;go[i]=999999999;
        for(int j=i+1;j<=n;j++)
        {
            int u=p[j-1];
            while(u && ch[u+1+tmp]!=ch[j])u=p[u+tmp];
            if(ch[u+1+tmp]==ch[j])u++;
            p[j]=u;
            go[j]=go[u+tmp];
            if(p[j]>=K)go[j]=min(go[j],p[j]);
            if(go[j]>=K && 2*go[j]<=(j-i+1)-1)ans++;
        }
        tmp++;
    }
    printf("%d\n",ans);
    return 0;
}

[bzoj3620][kmp]似乎在梦中见过的样子

猜你喜欢