KMP算法 C++版本 Python版本

AcWing 831. KMP字符串  https://www.acwing.com/problem/content/833/

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

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

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

输入格式

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

第二行输入字符串P。

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

第四行输入字符串S。

输出格式

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

数据范围

1≤N≤1041≤N≤104
1≤M≤1051≤M≤105

输入样例：

3
aba
5
ababa

输出样例：

0 2

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 1e5+5;

char p[N]; //n  小
char s[N]; //m  大
int ne[N];
int main()
{

    int n;
    cin >> n;
    for(int i = 1; i <= n; ++ i)
        cin >> p[i];
    int m;
    cin >> m;
    for(int i = 1; i <= m; ++ i)
        cin >> s[i];

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

    for(int i = 1, j = 0; i <= m; ++ i)
    {
        while(j && s[i] != p[j+1]) j = ne[j];
        if(s[i] == p[j+1])
            ++ j;
        if(j == n)
        {
            cout << i - n << " ";
            j = ne[j];
        }

    }
    return 0;
}

n = int(input())
p = [" "] + list(input())
m = int(input())
s = [" "] + list(input())
ne = [0 for i in range(n+5)]

j = 0
for i in range(2, n+1):
    while j and p[i] != p[j+1]:
        j = ne[j]
    if p[i] == p[j+1]:
        j += 1
    ne[i] = j
j = 0
for i in range(1, m+1):
    while j and s[i] != p[j+1]:
        j = ne[j]
    if s[i] == p[j+1]:
        j += 1
    if j == n:
        print(i - n, end=" ")
        j = ne[j]