洛谷P3375 【模板】KMP字符串匹配

KMP字符串匹配

题目描述

如题，给出两个字符串s1和s2，其中s2为s1的子串，求出s2在s1中所有出现的位置。

为了减少骗分的情况，接下来还要输出子串的前缀数组next。如果你不知道这是什么意思也不要问，去百度搜[kmp算法]学习一下就知道了。

输入输出格式

输入格式：
第一行为一个字符串，即为s1（仅包含大写字母）

第二行为一个字符串，即为s2（仅包含大写字母）

输出格式：
若干行，每行包含一个整数，表示s2在s1中出现的位置

接下来1行，包括length(s2)个整数，表示前缀数组next[i]的值。

分析：如题目，KMP模板。。。next[i]表示s2中以i结尾的非前缀子串与s2的前缀能够匹配的最长长度。

代码

#include <cstdio>
#include <cstring>
#include <string>
#define N 1000005
using namespace std;

char a[N],b[N];
int next[N];

int main()
{
    scanf("%s", a + 1);
    scanf("%s", b + 1);
    int la = strlen(a + 1);
    int lb = strlen(b + 1);
    int j = 0;
    for (int i = 2; i <= lb; i++)
    {
        while (j && b[j + 1] != b[i]) j = next[j];
        if (b[j + 1] == b[i]) j++;
        next[i] = j;
    }
    j = 0;
    for (int i = 1; i <= la; i++)
    {
        while (j && b[j + 1] != a[i]) j = next[j];
        if (b[j + 1] == a[i]) j++;
        if (j == lb) printf("%d\n", i - lb + 1);
    }
    for (int i = 1; i <= lb; i++)
        printf("%d ", next[i]);
}