P3375 【模板】KMP字符串匹配


链接

code

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cctype>
#include <vector>
#include <iostream>
#include <map>
#include <set>

using namespace std;
inline int read(){
    int x = 0, op = 1; char ch = getchar();
    while (!isdigit(ch)){
        if (ch == '-') op = -1; ch = getchar();}
    while (isdigit(ch)){
        x = (x << 1) + (x << 3) + (ch ^ 48);
        ch = getchar();
    }
    return x * op;
}
inline void write(int x){
    if (x < 0) putchar('-'), x = - x;
    if (x > 9) write(x / 10);
    putchar('0' + x % 10);
}

vector<int> getPi(string s){
    int n = s.length();
    vector<int> pi(n);
    for (int i = 1; i < n; ++i) {
        int j = pi[i - 1];
        while (j && s[i] != s[j]) j = pi[j - 1];
        if (s[i] == s[j]) j++;
        pi[i] = j;
    }
    return pi;
}

int main() {
#ifdef DEBUG
    freopen("in.txt", "r", stdin);
#endif
    string a, b;
    cin >> a >> b;
    vector<int> pi = getPi(b);

    int j = 0;
    for (int i = 0; i < a.length(); ++i) {
        while (j && b[j] != a[i]) j = pi[j - 1];
        if (b[j] == a[i]) j++;
        if (j == b.length()){
            cout << i - b.length() + 2 << endl;
            j = pi[j - 1];
        }
    }
    for (int e : pi) {
        cout << e << " ";
    }
}
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 护眼 设计师:闪电赇 返回首页