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

任重而道远

题目描述

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

为了减少骗分的情况，接下来还要输出子串的前缀数组next。

（如果你不知道这是什么意思也不要问，去百度搜[kmp算法]学习一下就知道了。）

输入输出格式

输入格式：

第一行为一个字符串，即为s1

第二行为一个字符串，即为s2

输出格式：

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

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

输入输出样例

输入样例#1： 

ABABABC
ABA

输出样例#1： 

1
3
0 0 1 

AC代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;

const int N = 1000010;
int len, idx = 0, ans[N], nxt[N];
string s1, s2;

void get () {
    nxt[0] = -1, nxt[1] = 0;
    for (int i = 2, j = 0; s2[i]; i++) {
        while (~j && s2[j + 1] != s2[i]) j = nxt[j];
        nxt[i] = ++j;
    }
}

void cmp () {
    for (int i = 0, j = 0; s1[i]; i++) {
        while (~j && s2[j + 1] != s1[i]) j = nxt[j];
        j++;
        if (j == len) {
            ans[++idx] = i - len + 2;
            j = nxt[j];
        }
    }
}

int main () {
    cin >> s1;
    cin >> s2;
    len = s2.size ();
    for (int i = len; i >= 1; i--) s2[i] = s2[i - 1];
    get (), cmp ();
    for (int i = 1; i <= idx; i++)
      printf ("%d\n", ans[i]);
    for (int i = 1; i <= len; i++)
      printf ("%d ", nxt[i]);
    return 0;
}