KMP算法

KMP字符串

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1000010;

int n,m;
char s[N],p[N];    
int ne[N];

int main()
{
    cin>>n>>p+1;  //下标从1开始
    cin>>m>>s+1;
    
    //求next[]数组(next[1]=0)
    for(int i=2,j=0;i<=n;i++)
    {
        while(j&&p[i]!=p[j+1]) j=ne[j];
        if(p[i]==p[j+1]) j++;
        ne[i]=j;
    }
    
    //kmp匹配过程
    for(int i=1,j=0;i<=m;i++)
    {
        while(j&&s[i]!=p[j+1]) j=ne[j];  //退无可退(j==0)/s[i]与p[j+1]不匹配
        if(s[i]==p[j+1]) j++;
        if(j==n)                        //匹配成功
        {
            cout<<i-n<<" ";             //由于题目要求下标从0开始((i-n+1)----->(i-n))
            j=ne[j];                    //继续匹配下一个位置
        }
    }
    return 0;
}