AcWing 831.KMP字符串

AcWing 831.KMP字符串

题目描述

给定一个字符串 S，以及一个模式串 P ，

所有字符串中只包含大小写英文字母以及阿拉伯数字。

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

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

输入格式

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

第二行输入字符串 P。

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

第四行输入字符串 S。

输出格式

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

数据范围

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

输入样例

3
aba
5
ababa

输出样例

0 2

题目思路

KMP模板题，next数组与数据数组都从1开始存储。
next数组求值方法:
我们能确定next数组第一位一定为0(第0位也为0)，后面求解每一位的next值时，根据前一位进行比较。
从第二位开始，将前一位与其next值对应的内容进行比较，
如果相等，则该位的next值就是前一位的next值加上1；
如果不等，向前继续寻找next值对应的内容来与前一位进行比较，
直到找到某个位上内容的next值对应的内容与前一位相等为止，
则这个位对应的值加上1即为需求的next值；
如果找到第一位都没有找到与前一位相等的内容，那么求解的位上的next值为1。

#include<cstdio>
#include<iostream>

using namespace std;

const int N = 1e5+10;
const int M = 1e6+10;

int n,m;
char p[N],s[M];
int ne[N];

int main()
{
    cin >> n >> p + 1 >> m >> s + 1;
    
    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;
}