P3375 [模板]KMP

点击查看代码

#include<bits/stdc++.h>

using namespace std;

const int N=1e6+10;
char s1[N],s2[N];
vector<int> pos;
int ne[N];

int main()
{
    scanf("%s%s",s1+1,s2+1);
    int n=strlen(s1+1);
    int m=strlen(s2+1);
    for(int i=2,j=0;i<=m;i++){
        while(j&&s2[j+1]!=s2[i]) j=ne[j];
        if(s2[j+1]==s2[i]) j++;
        ne[i]=j;
    }

    for(int i=1,j=0;i<=n;i++){
        while(j&&s2[j+1]!=s1[i]) j=ne[j];
        if(s2[j+1]==s1[i]) j++;
        if(j==m){
            pos.push_back(i-m+1);
            j=ne[j];
        }
    }

    for(auto p : pos){
        cout<<p<<endl;
    }
    for(int i=1;i<=m;i++){
        cout<<ne[i]<<" ";
    }

    return 0;
}

也是一道很经典的kmp模板题，不过通过这道题目，我加深了对于j从0开始，用j+1匹配的好处了 这样j永远代表匹配成功的长度，回退的时候会很优雅，直接j=ne[j]即可