洛谷 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;
}

 

    原文作者:KMP算法
    原文地址: https://blog.csdn.net/integrator_37/article/details/81707226
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞