bzoj 3942: [Usaco2015 Feb]Censoringkmp+栈

Posted 2020-11-03 lokiii

好久没写kmp都不会写了……
开两个栈，s存当前串，c存匹配位置
用t串在栈s上匹配，栈每次入栈一个原串字符，用t串匹配一下，如果栈s末尾匹配了t则弹栈

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1000005;
int n,m,c[N],top,ne[N];
char a[N],b[N],s[N];
void gtnxt()
{
    ne[0]=-1;
    for(int i=1,j=-1;i<n;i++)
    {
        while(j!=-1&&b[i]!=b[j+1]) 
            j=ne[j];
        if(b[i]==b[j+1]) 
            j++;
        ne[i]=j;
    }
}
int main()
{
    scanf("%s%s",a,b);
    n=strlen(a),m=strlen(b);
    gtnxt();
    for(int i=0;i<n;i++)
    {
        int j=c[top];
        s[++top]=a[i];
        while(j!=-1&&b[j+1]!=s[top])
            j=ne[j];
        if(b[j+1]==s[top])
            j++;
        c[top]=j;
        if(c[top]==m-1)
            top-=m;
    }
    s[top+1]=‘\0‘;
    puts(s+1);
    return 0;
}