KMP板子题

Posted 2020-11-16 dongdong25800

 

 

 

https://www.luogu.org/problem/P3375

 

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
int nxt[maxn];
int f[maxn];
char a[maxn];
char b[maxn];
int lena,lenb;
int ans[maxn],cnt;
void getnxt() // a是子串b是主串
{
    nxt[1]=0;
    for(int i=2,j=0; i<=lena; i++)
    {
        while(j>0&&a[i]!=a[j+1]) j=nxt[j];
        if(a[i]==a[j+1]) j++;
        nxt[i]=j;
    }
}
void kmp()
{
    int i,j;
    for(i=1,j=0; i<=lenb; i++)
    {
        while(j>0&&(j==lena||b[i]!=a[j+1])) j=nxt[j];
        if(b[i]==a[j+1]) j++;
        if(j==lena)
            ans[++cnt]=i-lena+1;
    }
}
int main()
{
    scanf("%s%s",b+1,a+1);
    lena=strlen(a+1);
    lenb=strlen(b+1);
    getnxt();
    kmp();
    for(int i=1; i<=cnt; i++)
        printf("%d
",ans[i]);
    for(int i=1; i<=lena; i++)
        printf("%d ",nxt[i]);
}