831. KMP字符串

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

模式串 P� 在字符串 S� 中多次作为子串出现。

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

输入格式

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

第二行输入字符串 P�。

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

第四行输入字符串 S�。

输出格式

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

数据范围

1≤N≤10^5
1≤M≤10^6

#include<iostream>

using namespace std;

const int N = 1004000;
char p[N],s[N];//s字符串，p模式串
//next[j]表示1-j内，字符串p[1-i]与p[m - i + 1,m]相等，next[j]存储的就是m,下次匹配会从j + 1开始
int nex[N];
int n,m;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    
    cin >> n >> p + 1 >> m >> s + 1;//下标从1开始
    //找next[]
    for(int i = 2, j = 0; i <= n; i ++)//j = next[j]后，下一次匹配是s[i]与p[j + 1]
    {
        //j存在且s[i] != p[j]
        while(j && p[i] != p[j + 1]) j = nex[j];
        //从next[j]下个位置开始匹配
        if(p[i] == p[j + 1]) j ++;
        nex[i] = j;
    }
    for(int i = 1, j = 0; i <= m; i ++)//j = next[j]后，下一次匹配是s[i]与p[j + 1]
    {
        //j存在且s[i] != p[j]
        while(j && s[i] != p[j + 1]) j = nex[j];
        //从next[j]下个位置开始匹配
        if(s[i] == p[j + 1]) j ++;
        if(j == n)
        {
            cout << i - n << " ";//下标从1开始
            j = nex[j];
        }
    }
    return 0;
}