洛谷_3375 KMP字符串匹配

题意

给出A，B两个字符串，求出B串在A串中出现的位置以及B串中的每一个位置的next（就是代表B串中以当前位置结尾的子串和B的前缀可以匹配的最大长度）。

思路

KMP的模板题。我们设两个指针i，j（其中i指A，j指B），如果可以扩展j的话我们就让j+1，否则我们让j=next[j]，把匹配的长度缩短，看能不能从这里继续扩展，直到j变成0后我们就重头开始匹配。求next的方法和这个类似，就是让B串和B串自己匹配。

代码

#include<cstdio>
#include<cstring>
#define maxn 1000001
using namespace std;
int next[maxn],l1,l2;
char s1[maxn],s2[maxn];
int main()
{
   scanf("%s%s",s1+1,s2+1);//相当于从1~n读入s，而不是从0~n-1
    l1=strlen(s1+1);
    l2=strlen(s2+1);
    int j=0;//j代表当前匹配的最大长度
    for (int i=2;i<=l2;i++)
    {
        while (j>0&&s2[i]!=s2[j+1]) j=next[j];//如果不能扩展我们就缩短
        if (s2[i]==s2[j+1]) j++;//如果可以扩展就扩展
        next[i]=j;
    }
    j=0;
    for (int i=1;i<=l1;i++)
    {
        while (j>0&&s1[i]!=s2[j+1]) j=next[j];
        if (s1[i]==s2[j+1]) j++;
        if (j==l2)//如果当前匹配的长度已经和B串的长度一样了就是匹配成功了
        {
            printf("%d\n",i-l2+1);//输出
            j=next[j];//后面可能还能匹配，所以更新
        }
    }
    for (int i=1;i<=l2;i++)
        printf("%d ",next[i]);
}