记录一个KMP模板

洛谷有题

/*说明：s1为主串，s2为模式串。(这字符串的下标都是从1开始)
输出的是模式串在主串每次成功匹配的开始位置。
最后一行输出的是next数组
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
#define ll long long
char s1[N],s2[N];
int net[N];
void getnet(int len)
{
    int i=2,j=0;
    net[1]=0;
    for(int i=2,j=0; i<=len; i++)
    {
        while(j>0&&s2[i]!=s2[j+1]) j=net[j];
        if(s2[i]==s2[j+1])
        {
            j++;
        }
        net[i]=j;
    }
}
void kmp(int l1,int l2)
{
    int f;
    for(int i=1,j=0; i<=l1; i++)
    {
        f=0;
        while(j>0&&(j==l2||s1[i]!=s2[j+1]))j=net[j];
        if(s1[i]==s2[j+1])j++;
        f=j;
        if(f==l2)
        {
            //j=0;     不能有重复的就加上
            printf("%d\n",i-f+1);
        }
    }
}
int main()
{
    scanf("%s%s",s1+1,s2+1);
    int l1=strlen(s1+1);
    int l2=strlen(s2+1);
    getnet(l2);
    kmp(l1,l2);
    for(int i=1; i<=l2; i++)
    {
        printf("%d ",net[i]);
    }
    printf("\n");
    return 0;
}