kmp算法

1、思路

ne初始化代码模板：

ne[1]=0;//不用写
for(int i=2,j=0;i<=n;i++)//n是p串的长度，i从1开始，j从0开始，第一位ne一定为0，从第二位开始初始化
{
    while(j&&p[i]!=p[j+1])//当j不为0且下一位不匹配
        j=ne[j];//返回之前匹配的位置再进行比较
    if(p[i]==p[j+1])//当这一位匹配时
        j++;//长度加1
    ne[i]=j;//记录下此时的长度
}

for(int i=1,j=0;i<=n;i++)
{
    while(j&&s[i]!=p[j+1])//如果j有对应p串的元素，且下一位不匹配，需要移动
        j=ne[j];          //根据当前ne数组移动
    //用while是由于移动后可能仍然失配，所以要继续移动直到匹配或整个p串移到后面（j = 0)
    if(s[i]==p[j+1])//当前位匹配时，
        j++;
    //当前元素匹配，j移向p串下一位
    if(j==m)//当全部匹配时
    {
        //匹配成功，进行相关操作
        j = ne[j];  //继续匹配下一个子串
    }
}

完整代码模板：

#include<iostream>

using namespace std;

const int N=1e5+10;
const int M=1e6+10;

int n,m;//n是p串长度，m是s串长度
char s[M],p[N];
int ne[N];

int main()
{
    scanf("%d %s %d %s",&n,p+1,&m,s+1);//两个字符串都从1开始输入
    for(int i=2,j=0;i<=n;i++)//n是p串的长度，i从1开始，j从0开始，第一位ne一定为0，从第二位开始初始化
    {
        while(j&&p[i]!=p[j+1])//当j不为0且下一位不匹配
            j=ne[j];            //返回之前匹配的位置再进行比较
        if(p[i]==p[j+1])        //当这一位匹配时
            j++;                //长度加1
        ne[i]=j;                //记录下此时的长度
    }
    for(int i=1,j=0;i<=m;i++)
    {
        while(j&&s[i]!=p[j+1])//如果j有对应p串的元素，且下一位不匹配，需要移动
            j=ne[j];          //根据当前ne数组移动
                            //用while是由于移动后可能仍然失配，所以要继续移动直到匹配或整个p串移到后面（j = 0)
        if(s[i]==p[j+1])//当前元素匹配，j移向p串下一位
            j++;
        if(j==n)//当全部匹配时
        {
            printf("%d ",i-n);
            j = ne[j];      //继续匹配下一个子串
        }
    }
    return 0;
}