KMP二三事

KMP那些事

众所周知，kmp是一个非常强大的数据结构
她的作用有：快速匹配字符串，以及关于循环节的一系列事。
下面主要介绍这两个功能

快速匹配字符串

咱们看这么两个字符串，分别为S，P串，用P去匹配S
基本步骤为：

1.先处理模板串p的ne[]数组，ne[i] 表示 p 的1~i 子串最大的前缀与后缀相等的位置如：
ne[1] = 0, ne[2] = 0, ne[3] = 0, ne[4] = 1, ne[5] = 2, ne[6] = 3, ne[7] = 0, ne[8] = 0;
2.ne数组的作用： p的红色与S的红色已经匹配成功，但是第6个位置S是e，p是c匹配失败，我们前5个已经匹配上了，如果重新匹配就很浪费时间，我们这里的ne[5]=2表示前缀1 ~ 2与后缀4 ~ 5相等，所以我们就可以直接从p串3的位置再次开始与s串匹配，从而达到优化的目的

例题1

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10, M = 1e6 + 10;
int p[N], s[M];
int n, m, t;
int ne[N];

int main()
{
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d %d", &m, &n);
        for(int i = 1; i <= m; i++) scanf("%d", &s[i]);

        for(int i = 1; i <= n; i++) scanf("%d", &p[i]);
        
        for(int i = 2, j = 0; i <= n; i++)
        {
            while(j && p[j + 1] != p[i]) j = ne[j];
            if(p[j + 1] == p[i]) j++;
            ne[i] = j;
        }
        
        bool f = 1;
        for(int i = 1, j = 0; i <= m; i++)
        {
            while(j && p[j + 1] != s[i]) j = ne[j];
            if(p[j + 1] == s[i]) j ++;
            if(j == n)
            {
                f = 0;
                // cout << i << " "  << j << endl;
                printf("%d\n", i - n + 1);
                break;
            }
        }
        if(f) puts("-1");
                            
    }
    return 0;
}

例题2

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10, M = 1e6 + 10;
char p[N], s[M];
int n, m, t;
int ne[N];

int main()
{
    scanf("%d", &t);
    while(t--)
    {
        scanf("%s%s",p+1, s+1);

        for(int i = 2, j = 0; i <= (int)strlen(p+1); i++)
        {
            while(j && p[j + 1] != p[i]) j = ne[j];
            if(p[j + 1] == p[i]) j++;
            ne[i] = j;
        }

        LL ans = 0;
        for(int i = 1, j = 0; i <= (int)strlen(s+1); i++)
        {
            while(j && p[j + 1] != s[i]) j = ne[j];
            if(p[j + 1] == s[i]) j ++;
            if(j == (int)strlen(p+1))
            {
                ans ++;
                j = ne[j];
            }
        }
        
        printf("%lld\n", ans);                    
    }
    return 0;
}

最小循环节

最小循环节：L = len - ne[len]
因为 ne[len] 是最大的前缀与后缀相等，所以此时循环节是最小的

类型1：添加字符构成周期性

1.当 L 能整除 len 且 不等于 len 时： 则表示此时串为循环串 具有周期性
2.当 L 能整除 len 但 等于 len 时： 则表示串里面不存在循环节，或者说唯一的循环节就是她本身
3.当 L 不能整除 len 时： 后面还需要补上 ==L-（len%L）==个字符来使其构成周期性

类型2：循环节出现的次数

当len%L == 0时才表示该串为周期串，才存在循环节这一说，循环节出现的次数为：len / L

找出所有既是前缀又是后缀的子串

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10, M = 1e6 + 10;
char p[M], s[M];
int n, m, t;
int ne[M];

int main()
{
    while(~scanf("%s", p+1))
    {
        n = strlen(p+1);

        for(int i = 2, j = 0; i <= n; i++)
        {
            while(j && p[j + 1] != p[i]) j = ne[j];
            if(p[j + 1] == p[i]) j++;
            ne[i] = j;
        }
        
        vector<int> ans;
        while(n)
        {
            ans.push_back(n);
            n = ne[n];
        }
        
        for(int i = ans.size()-1; i >= 0; i --) printf("%d ", ans[i]);
        puts("");
        
    }
    return 0;
}