BZOJ3620: 似乎在梦中见过的样子 kmp

最新推荐文章于 2019-09-23 22:02:48 发布

_xgcxgc

最新推荐文章于 2019-09-23 22:02:48 发布

阅读量242

点赞数

CC 4.0 BY-SA版权

分类专栏： kmp xgc的做题记录文章标签： kmp

本文链接：https://blog.youkuaiyun.com/xgc_woker/article/details/80059056

xgc的做题记录同时被 2 个专栏收录

168 篇文章

订阅专栏

kmp

2 篇文章

订阅专栏

本文介绍了一种利用KMP算法解决寻找形如A+B+A字符串的问题，其中A和B为子串，A长度大于等于k，B长度大于等于1。通过枚举右端点并使用KMP算法来高效地计算出符合条件的子串数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description
求有多少个所有形似于A+B+A的字串,且len(A)>=k,len(B)>=1。

Sample Input
aaaaa
1

Sample Output
6

枚举右端点，做kmp，然后就OK了。
感觉kmp这东西好烦啊，写代码都不是很会写，每次都要想半天。

#include <cstdio>
#include <cstring>

using namespace std;

char ss[16000], s1[16000];
int kk, len, ans, p[16000];

int main() {
    scanf("%s", ss + 1);
    scanf("%d", &kk);
    len = strlen(ss + 1);
    for(int i = 1; i <= len; i++) {
        p[1] = 0;
        for(int j = 2; j <= len - i + 1; j++) {
            int k = p[j - 1];
            while(k && ss[k + i] != ss[j + i - 1]) k = p[k];
            if(ss[k + i] == ss[j + i - 1]) k++;
            p[j] = k;
        }
        int k = 0;
        for(int j = kk + 2; j <= len - i + 1; j++) {
            while(k && ss[k + i] != ss[j + i - 1]) k = p[k];
            if(ss[k + i] == ss[j + i - 1]) k++;
            while(j - k * 2 < 1) k = p[k];
            if(k >= kk && j - k * 2 >= 1) ans++;
        }
    }
    printf("%d\n", ans);
    return 0;
}