KMP算法 模板

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