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]);
}