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

最新推荐文章于 2020-07-30 09:50:57 发布

ws_fqk

最新推荐文章于 2020-07-30 09:50:57 发布

阅读量728

点赞数

分类专栏： My Code

本文链接：https://blog.csdn.net/Phenix_2015/article/details/50933086

版权

My Code 专栏收录该内容

349 篇文章 0 订阅

订阅专栏

枚举左端点跑KMP， $O(n^2)$ 也能过真是！@#￥%……

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,ans,k;
char s[20005];
int next[20005];
inline void KMP(int p)
{
    for (int i=1;i<=n;i++)
        next[i]=p-1;
    int j=p-1;
    for (int i=p+1;i<=n;i++)
    {
        while (j!=p-1&&s[j+1]!=s[i]) j=next[j];
        if (s[j+1]==s[i]) j++;
        next[i]=j;
    }
    j=p-1;
    for (int i=p+1;i<=n;i++)
    {
        while (j!=p-1&&s[j+1]!=s[i]) j=next[j];
        if (s[j+1]==s[i]) j++;
        while ((j-p+1)*2>=(i-p+1)) j=next[j];
        if (j-p+1>=k) ans++;
    }
}
int main()
{
    scanf("%s%d",s+1,&k);
    n=strlen(s+1);
    for (int i=1;i<=n;i++)
        KMP(i);
    cout << ans << endl;
}