数据结构与算法实验4——字符串和数组 7-2 【模板】KMP字符串匹配

7-2 【模板】KMP字符串匹配 分数 20

作者 李廷元

单位 中国民用航空飞行学院

给出两个字符串text和pattern，其中pattern为text的子串，求出pattern在text中所有出现的位置。

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

输入格式:

第一行为一个字符串，即为text。

第二行为一个字符串，即为pattern。

输出格式:

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

接下来1行，包括length(pattern)个整数，表示前缀数组next[i]的值，数据间以一个空格分隔，行尾无多余空

输入格式:

第一行为一个字符串，即为text。

第二行为一个字符串，即为pattern。

输出格式:

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

接下来1行，包括length(pattern)个整数，表示前缀数组next[i]的值，数据间以一个空格分隔，行尾无多余空格。

样例">输入样例:

ABABABC
ABA

输出样例:

1
3
0 0 1

 

样例说明:

 

代码长度限制

16 KB

时间限制

4000 ms

内存限制

64 MB

#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 5;
string S, T;
int Next[N];
void get_next()
{
    int i = 0, j = -1;
    Next[i] = -1;
    while (i < T.size())
    {
        if (j == -1 || T[i] == T[j])
        {
            Next[++i] = ++j;
        }
        else
            j = Next[j];
    }
}

int kmp(int pos)
{
    int i = pos, j = 0, ans = 0;
    while (i < S.size())
    {
        if (S[i] == T[j] || j == -1)
            ++i, ++j;
        else
            j = Next[j];
        if (j == T.size())
        {
            ans++;
            cout << i-j+1 << endl;
            //j = Next[j - 1];
            //i--;
            j=Next[j];
        }
    }
    return ans;
}

int main()
{
    cin >> S;
    cin >> T;
    get_next();
    kmp(0);
    for(int i=1; i<T.size()+1; i++)
    {
        if(i>1) cout << " ";
        cout << Next[i];
    }
    return 0;
}