3942: [Usaco2015 Feb]Censoring|Kmp

维护一个栈,一直插入元素如果匹配完了T串就删掉,记录栈中的每一个元素匹配到了T串的什么位置,这样复杂度就是 O(n)
还有一个卡常的小技巧,由于 puts 会比较快就在最后串位置上加上一个结束符,然后输出直接 puts

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<cstdio>
#include<queue>
#include<cmath>
#include<set>
#include<map>
#define N 1000005
using namespace std;
char a[N],b[N],st[N];
int La,Lb,top,now;
int pre[N],wh[N];
int main()
{
    scanf("%s%s",a+1,b+1);
    La=strlen(a+1);
    Lb=strlen(b+1);
    for(int i=2;i<=Lb;i++)
    {
        while(now&&b[now+1]!=b[i])now=pre[now];
        if(b[now+1]==b[i])now++;
        pre[i]=now;
    }
    for(int i=1;i<=La;i++)
    {
        int now=wh[top];
        st[++top]=a[i];
        while(now&&b[now+1]!=a[i])now=pre[now];
        if(b[now+1]==a[i])now++;
        if(now==Lb)top=top-Lb;
        else wh[top]=now;
    }
    st[top+1]='\0';
    puts(st+1);
    return 0;
}

你可能感兴趣的:(KMP)