洛谷 P3375 KMP字符串匹配(kmp模板题)
题目大意:
给出两个串A B,要求在A中找到B串,依次输出B在A中出现的首位置,之后输出一行b,bi 表示B串中前 i 个字符组成的串最长公共前后缀。
解题思路:
标准的kmp模板题,进行多次匹配,只需要改动一下kmp函数即可,如果j == plen, 则 j = nextval[j] 继续右移进行匹配。而对于最长公共前后缀,实质上就是从下标从 1 开始输出next 数组。
Code:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <cstring>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int N = 1e6 + 50;
int nextval[N];
string s, a;
void get_next()
{
int j = 0, k=-1;
nextval[0] = -1;
int slen = a.size();
while (j < slen)
{
if (k == -1 || a[j] == a[k])
{
j++; k++;
nextval[j] = k;
}
else k = nextval[k];
}
}
void kmp()
{
int i = 0, j=0;
int slen = s.size(), plen = a.size();
while (i < slen && j < plen)
{
if(j == -1 || s[i] == a[j])
{
i++;
j++;
if (j == plen)//这里改动一下,如果匹配到了继续右移
cout << i - j + 1 << endl, j = nextval[j];
}
else j = nextval[j];
}
}
int main()
{
ios::sync_with_stdio(false);
cin >> s >> a;
int len = a.size();
get_next();
kmp();
for (int i = 1; i <= a.size(); i ++)
cout << nextval[i] << ' ';
cout << endl;
return 0;
}