1. 程式人生 > >Censoring(棧+KMP)

Censoring(棧+KMP)

() name scanf bit print main spa 算法 scan

# 10048. 「一本通 2.2 練習 4」Censoring

【題目描述】

給出兩個字符串 $S$ 和 $T$,每次從前往後找到 $S$ 的一個子串 $A=T$ 並將其刪除,空缺位依次向前補齊,重復上述操作多次,直到 $S$ 串中不含 $T$ 串。輸出最終的 $S$ 串。

【算法】

1、kmp $O(n)$就可以定位。
2、棧是個好東西啊。
註:一開始想雙指針,實在不好寫。。。棧很好,它可以彈出去。。。(此處滑稽臉)

【代碼】

#include <bits/stdc++.h>
using namespace std;
int l1,l2,top;
int nxt[1000100],f[1000100];
char s[1000100],t[1000100],ans[1000100];
void parse() {
    nxt[1]=0;
    for(int i=2,j=0;i<=l2;i++) {
        while(j>0&&(j==l2||t[i]!=t[j+1])) j=nxt[j];
        if(t[i]==t[j+1]) j++;
        nxt[i]=j;
    }
}
int main() {
    scanf("%s%s",s+1,t+1);
    l1=strlen(s+1),l2=strlen(t+1);
    parse();
    for(int i=1,j=0;i<=l1;i++) {
        ans[++top]=s[i];
        while(j>0&&(j==l2||s[i]!=t[j+1])) j=nxt[j];
        if(s[i]==t[j+1]) j++;
        f[top]=j;
        if(j==l2) {
            top-=l2;
            j=f[top];
        }
    }
    ans[++top]='\0';
    printf("%s\n",ans+1);
    return 0;
}

Censoring(棧+KMP)