KMP模板
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 100005;
char t[maxn],p[maxn];
int lent,lenp;
int next[maxn];
int h=0;//数组从0开始就赋值为0,从1开始就赋值为1,等等,t数组和p数组要统一
void createNext()
{
int j=h-1;
next[h]=j;
for(int i=h+1;i<h+lenp;i++){
while(j>h-1&&p[j+1]!=p[i]){
j=next[j];
}
if(p[j+1]==p[i]){
++j;
}
next[i]=j;
}
/*for(int i=h;i<h+lenp;i++){
printf("next%d: %d\n",i-h+1,next[i]);
}*/
}
int KMP()
{
int num=0;//统计出现次数
int j=h-1;
for(int i=h;i<h+lent;i++){
while(j>h-1&&p[j+1]!=t[i]){
j=next[j];
}
if(p[j+1]==t[i]){
++j;
}
if(j==lenp+h-1){
j=next[j];// ***
num++;
return i-lenp+1;//模式串在文本串中第一次出现的位置
}
}
if(num==0){ //如果不存在就会返回-1,要想返回出现次数就把这个if和上面的return注释掉
return -1;
}
//return num;//返回模式串在文本串中出现的次数(允许部分重叠,如果要不允许重叠,把注释有***的那一行改为j=h-1)
}
int main(void)
{
gets(t);
gets(p);
lenp=strlen(p);//不是从0开始的记得改成 p+开始的位置
lent=strlen(t);
createNext();
printf("pos: %d\n",KMP());
return 0;
}
一道例题的应用:CodeForces-gym101466E-Text Editor
AC代码:
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 100005;
char t[maxn],p[maxn];
int lent,lenp;
int next[maxn];
int h=0;//数组从0开始就赋值为0,从1开始就赋值为1,等等,t数组和p数组要统一
void createNext()
{
int j=h-1;
next[h]=j;
for(int i=h+1;i<h+lenp;i++){
while(j>h-1&&p[j+1]!=p[i]){
j=next[j];
}
if(p[j+1]==p[i]){
++j;
}
next[i]=j;
}
/*for(int i=h;i<h+lenp;i++){
printf("next%d: %d\n",i-h+1,next[i]);
}*/
}
int KMP(int len)
{
int num=0;//统计出现次数
int j=h-1;
for(int i=h;i<h+lent;i++){
while(j>h-1&&p[j+1]!=t[i]){
j=next[j];
}
if(p[j+1]==t[i]){
++j;
}
if(j==len-1){
j=next[j];// ***
num++;
//return i;//模式串在文本串中第一次出现的位置,如果不存在会返回h+lent
}
}
return num;//返回模式串在文本串中出现的次数(允许部分重叠,如果要不允许重叠,把注释有***的那一行改为j=h-1)
}
int main(void)
{
gets(t);
gets(p);
//scanf("%s%s",t,p);//有空格记得换成gets
lenp=strlen(p);//不是从0开始的记得改成 p+开始的位置
lent=strlen(t);
createNext();
int k;
scanf("%d",&k);
int l=1,r=lenp;
int ans=-1;
while(l<=r){
int mid=(l+r)>>1;
if(KMP(mid)>=k){
ans=mid;
l=mid+1;
}
else{
r=mid-1;
}
}
if(ans==-1){
printf("IMPOSSIBLE\n");
}
else{
for(int i=0;i<ans;i++){
printf("%c",p[i]);
}
printf("\n");
}
return 0;
}