AcWing 831. KMP字符串

AcWing 831. KMP字符串

题目描述

给定一个模式串 S，以及一个模板串 P，所有字符串中只包含大小写英文字母以及阿拉伯数字。

模板串 P 在模式串 S 中多次作为子串出现。

求出模板串 P 在模式串 S 中所有出现的位置的起始下标。

输入格式

第一行输入整数 N，表示字符串 P 的长度。

第二行输入字符串 P。

第三行输入整数 M，表示字符串 S 的长度。

第四行输入字符串 S。

输出格式

共一行，输出所有出现位置的起始下标（下标从 0 开始计数），整数之间用空格隔开。

数据范围

$1\le N\le 10^5,1\le M\le 10^6$

输入样例：

3
aba
5
ababa

输出样例：

0 2

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 2e5 + 5;
int _next[N];
string s1, s2;//这里的模板s1是文本串s2是匹配串(模式串)
void build_next(int len) {//对模式串进行预处理求以某个点为终点的最长公共前后缀
    _next[0] = -1;//0的时候为0，即从1开始(可能会错位)
    for (int i = 0, j = -1; i < len;) {
        if (j == -1 || s2[i] == s2[j]) {//如果说回到起始位置或者匹配上了
            ++j;++i;//都往前移动一位
            _next[i] = j;//记录下最长公共前后缀的长度
        }
        //如果说当前位置没有匹配上，匹配串移动距离为_next[j]
        else j = _next[j];
    }
}
void KMP(int len1, int len2) {
    build_next(len2);
    for (int i = 0, j = 0;i < len1;) {
        //如果说回到起始位置或者匹配上了，则匹配后一位
        if (j == -1 || s1[i] == s2[j]) ++i,++j;
        //如果说当前位置没有匹配上，匹配串移动距离为_next[j]
        else j = _next[j];
        //当匹配串匹配到末尾则说明匹配成功
        if (j == len2) {//匹配成功之后的逻辑
            j = _next[j];
            cout << i - len2 <<" ";
        }
    }
}
int main() {
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    
    int n,m;
    cin >> n >> s2 >> m >> s1;
    KMP(m, n);


    return 0;
}

考研版本kmp新理解

int _next[N];
void get_next(string p) {
    _next[0] = -1;//第一个位置默认其为-1
    int i = 0, j = -1;//位置差一前后缀才能开始比较
    while (i < p.length()) {
        if (j == -1 || p[i] == p[j]) {
            //第一种情况是 第一个字符没有匹配上 默认其为0(从第二个开始统一化)
            //第二种情况是 匹配上了 则最长公共前后缀加1
            _next[++i] = ++j;
        }
        else j = _next[j];
    }
}
void kmp(string s, string p) {
    get_next(p);
    int i = 0, j = 0;//数组下标从0开始
    while (i < s.length()) {//如果只匹配一次则可以加 &&j<p.length()
        if (j == -1 || s[i] == p[j]) {
            //第一种情况是模式串的 第一个字符 没有与主串匹配上则 继续向后匹配
            //第二中情况是对应字符匹配上了同样 继续向后匹配
            i++, j++;
        }
        else j = _next[j];
        if (j == p.length()) {//匹配为后的逻辑
            cout << i - j + 1 << "\n";
            j = _next[j];
        }
    }
}