【模板】KMP字符串匹配

终于搞懂KMP了！QAQ

教材：KMP字符串匹配算法2_哔哩哔哩 (゜-゜)つロ 干杯~-bilibili

然后根据自己的理解写的代码↓

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXN=1000007;
char s[MAXN],t[MAXN];
int nxt[MAXN];
int main()
{
    cin>>s>>t;
    int lens=strlen(s),lent=strlen(t);
    //make_nxt
    int len=0;
    int i=1;
    nxt[0]=0;
    while(i<lent){
        if(t[i]==t[len]){
            len++;
            nxt[i]=len;
            i++;
        }
        else{
            if(len){
                len=nxt[len-1];
            }
            else{
                nxt[i]=0;
                i++;
            }
        }
    }
    //move_nxt
    for(int i=lent;i;--i)
    nxt[i]=nxt[i-1];
    nxt[0]=-1;
    //search
    int j=0;
    i=0;
    while(i<lens){
        if(j==lent-1){
            if(s[i]==t[j]){
                cout<<i-j+1<<endl;
                j=nxt[j];
            }
        }
        if(s[i]==t[j]){
            i++;j++;
        }
        else{
            if(j!=-1){
                j=nxt[j];
            }
            else{
                i++;j++;
            }
        }
    }
    //print
    for(int i=1;i<=lent;++i)
    cout<<nxt[i]<<" ";
    return 0;
}

以及 这个代码确实很不优

以前从网上抄的但根本写不出来的快很多的代码↓

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
char s1[1000007];
char s2[1000007];
int nex[1000007];
int main()
{
    //freopen("a.in","r",stdin);
    scanf("%s\n%s",&s1,&s2);
    int p,k,i,j,len1=strlen(s1),len2=strlen(s2);
    k=-1;
    nex[0]=-1;
    for(p=1;p<len2;++p)
    {
        while(k>-1&&s2[k+1]!=s2[p])k=nex[k];
        if(s2[k+1]==s2[p])k++;
        nex[p]=k;
    }
    k=-1;
    for(p=0;p<len1;++p)
    {
        while(k>-1&&s2[k+1]!=s1[p])k=nex[k];
        if(s2[k+1]==s1[p])k++;
        if(k==len2-1){printf("%d\n",p-len2+2);k=-1;p=p-len2+1;}
    }
    for(i=0;i<len2;++i)
    printf("%d ",nex[i]+1);
    return 0;
}

其实里面的思路都差不多，但是在模板题「LuoguP3375」【模板】KMP字符串匹配 上测，有差不多翻倍的时间差距。

不过用for这样搞确实没有while好想好打（逃

 

18.10.6 UPD:

今天考试 有30分给的KMP

我硬是没想起来自己理解的超长代码，随手背出了网上抄的那个

真香