[模板] KMP

被各路奇怪姿势模板洗脑后,决定自己用自己的理解写个模板。

特地用[codevs1204]测了下保证准确性。

时间复杂度:O(n)

#include<set> 
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<cstdio>
#include<vector> 
#include<string>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 1000001
using namespace std;
int next[N],m,n;
char a[N],b[N];
inline void get_next(){
    for(int i=2,j=0;i<=m;i++){
        while(j&&b[i]!=b[j+1]) j=next[j];
        j+=(b[i]==b[j+1]);
        next[i]=j;
    }
}
inline void kmp(){
    for(int i=1,j=0;i<=n;i++){
        while(j&&a[i]!=b[j+1]) j=next[j];
        if(a[i]==b[j+1]) j++;
        if(j==m) printf("%d\n",i-m+1);
    }
}
inline void init(){
    scanf("%s%s",a+1,b+1);
    n=strlen(a+1);
    m=strlen(b+1);
    get_next();
    kmp();
}
int main(){
    freopen("kmp.in","r",stdin);
    freopen("kmp.out","w",stdout);
    init();
    fclose(stdin);
    fclose(stdout);
    return 0;
}  
原文地址:https://www.cnblogs.com/AireenYe/p/KMP.html