AcWing 831. KMP字符串

问题描述

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

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

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

输入格式：

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

第二行输入字符串$P$。

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

第四行输入字符串$S$。

输出格式：

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

注意，结果保留$6$位小数。

数据范围

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

输入样例：

3
aba
5
ababa

输出样例：

0 2

思路

1、使用KMP算法，进行字符串的匹配。
2、KMP算法主要就是用来解决字符串的匹配问题。

原题链接

#include <iostream>

using namespace std;

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

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

int main()
{
    cin >> n >> p + 1 >> m >> s + 1;
    
    //求ne过程 //i从2 开始 ne 1 = 0 第一个失败了只能从0开始
    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 ++ ) //匹配 i 从1 开始j 从0 开始 j与i错一位
    {
        while (j && s[i] != p[j + 1]) //j没有退回起点
        {
            j = ne[j]; //匹配为止  或者退无可退了 退一步
        }
        if (s[i] == p[j + 1]) j ++; //匹配成功
        if (j == n)
        {
            //匹配成功
            printf("%d ", i - n);
            j = ne[j];//匹配成功退一步
        }
        
    }
    return 0;
}