模板汇总——KMP & EX-KMP

1. kmp

　　相当于往前求出一段字符信息，使得 这段字符信息和前缀相等。

　　　

void getnext(){
    int k = -, j = ;
    nx[] = -;
    while(j < m){
        if(k == - || b[j] == b[k]) nx[++j] = ++k;
        else k = nx[k];
    }
}

例：

T

#include<iostream>
using namespace std;
int nx[+], b[+], a[+];
int n, m;
void getnext(){
    int k = -, j = ;
    nx[] = -;
    while(j < m){
        if(k == - || b[j] == b[k]) nx[++j] = ++k;
        else k = nx[k];
    }
}
int KMP()
{
    int flag = -;
    for(int i = , j = ; i < n; i++){
        while(j >  && b[j] != a[i]) j = nx[j];
        if(a[i] == b[j]) j++;
        if(j == m)
        {
            return i - j + ;
        }
    }
    return -;
}
int main()
{
    ios::sync_with_stdio();
    cin.tie();
    cout.tie();
    int t;
    cin >> t;
    while(t--){
        cin >> n >> m;
        for(int i = ; i < n; i++) cin >> a[i];
        for(int i = ; i < m; i++) cin >> b[i];
        getnext();
        cout << KMP() << endl;
    }
    return ;
}

2. ex-kmp

　　相当于往后求出一段信息，使得这段信息和前缀相等。

int n, z[N];
char s[N];
void init(){
    z[] = n;
    int j = , k;
    for(int i = ; i < n; i = k){
        if(j < i) j = i;
        while(j < n && S[j] == S[j-i]) j++;
        z[i] = j-i;
        k = i+;
        while(k + z[k-i] < j)
            z[k]=z[k-i],k++;
    }
}

例：

T：

给出模板串A和子串B，长度分别为lenA和lenB，要求在线性时间内，对于每个A[i](1<=i<=lenA)，求出A[i..lenA]与B的最长公共前缀长度。

#include<bits/stdc++.h>
using namespace std;
const int N = 2e6+;

char s[N], ss[N];
int z[N];
void init(int n){
    z[] = n;
    int j = , k;
    for(int i = ; i < n; i = k){
        if(j < i) j = i;
        while(j < n && s[j] == s[j-i]) j++;
        z[i] = j-i;
        k = i+;
        while(k + z[k-i] < j)
            z[k]=z[k-i],k++;
    }
}
int Ac(){
    scanf("%s", ss);
    scanf("%s", s);
    int n = strlen(s), m = strlen(ss);
    s[n] = '@';
    strcpy(s++n, ss);
    init(n+m+);
    for(int i = n+; i <= n+m; ++i){
        printf("%d%c", z[i], " \n"[i==n+m]);
    }
    return ;
}
int main(){
    Ac();
    return ;
}