题目描述
如题,给出两个字符串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
说明
时空限制:1000ms,128M
数据规模:
设s1长度为N,s2长度为M
对于30%的数据:N<=15,M<=5
对于70%的数据:N<=10000,M<=100
对于100%的数据:N<=1000000,M<=1000000
样例说明:
所以两个匹配位置为1和3,输出1、3
//
// 洛谷P3375-KMP.cpp
// Cheese
//
// Created by 廖启帆 on 2019/4/4.
// Copyright © 2019 廖启帆. All rights reserved.
//
/*
ABABABC
ABA
1
3
0 0 1
*/
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int kmpcount(char text[],char pattern[]);
void getnext(char s[],long len);
int nexts[1000100]={0};
char text[1000100],pattern[1000100];
int main()
{
int i;
long len;
cin>>text>>pattern;
len=strlen(pattern);
kmpcount(text, pattern);
for (i=0; i<len; i++) {
printf("%d ",nexts[i]+1);
}
return 0;
}
int kmpcount(char text[],char pattern[]) { //统计pattern在text中出现的次数
int i,j=-1,count=0; //count表示成功匹配次数
long n=strlen(text),m=strlen(pattern); //字符串长度
getnext(pattern, m); //计算pattern的nexts数组
for (i=0; i<n; i++) { //试图匹配text[i]
while (j!=-1&&text[i]!=pattern[j+1]) { //直到j回到-1或text[i]==pattern[j+1]
j=nexts[j]; //不断回退
}
if (text[i]==pattern[j+1]) { //text[i]与pattern[j+1]匹配成功
j++; //令j+1
}
if (j==m-1) { //pattern完全匹配,说明pattern是text的子串
count++; //成功匹配次数+1
cout<<i-m+2<<endl;
j=nexts[j]; //让j回退到nexts[j]继续匹配
}
}
return count; //返回成功匹配次数
}
void getnext(char s[],long len) { //求解长度为len的字符串s的next数组
int i,j;
nexts[0]=-1; //nexts[i]为子串s[0~i]的最长相等前后缀最后一位的下标
j=nexts[0]; //初始化j=nexts[0]=-1
for (i=1; i<len; i++) { //求解nexts[1]~nexts[len-1]
while (j!=-1&&s[i]!=s[j+1]) { //直到j回退到-1或是s[i]==s[j+1]
j=nexts[j]; //反复令j=nexts[j]
}
if (s[i]==s[j+1]) { //如果s[i]==s[j+1]
j++; //则nexts[i]=j+1,先令j指向这个位置
}
nexts[i]=j; //令nexts[i]=j
}
}