3942: [Usaco2015 Feb]Censoring

题目链接

题目大意：给定两个串A和B，要求将A中删掉所有的B后输出

题解：求B的next数组，将A依次入栈，跑kmp

我的收获：……

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int N=1000100;

int n,m,top;
int nex[N],sta[N],a[N];
char S[N],T[N];

void getnext(){
    for(int i=2,j=0;i<=m;i++){
        while(j&&T[i]!=T[j+1]) j=nex[j];
        if(T[i]==T[j+1]) j++;
        nex[i]=j;
    }
}

void work()
{
    for(int i=1;i<=n;i++){
        sta[++top]=S[i];
        int j=a[top-1];
        while(j&&sta[top]!=T[j+1]) j=nex[j];
        if(sta[top]==T[j+1]) j++;
        a[top]=j;
        if(a[top]==m) top-=m;  
    }
    for(int i=1;i<=top;i++) putchar(sta[i]);
    putchar('\n');
} 

void init()
{
    scanf("%s%s",S+1,T+1);
    n=strlen(S+1),m=strlen(T+1);
    getnext();
}

int main()
{
    init();
    work();
    return 0;
}