1. 程式人生 > 其它 >【力扣】第 284 場周賽

【力扣】第 284 場周賽

P3121 [USACO15FEB]Censoring G

怎麼說?這題還算是比較妙的。

注意到題目中 FJ 每次在 s中找到最早出現的列表中的單詞(最早出現指該單詞的開始位置最小),然後從s中刪除這個單詞 這句話,所以我們對文字串跑自動機的時候,記錄一下刪掉串的長度和刪了這個串頂上的那個點的編號,開雙棧,一個存字元,最後輸出答案;另一個存節點編號,方便回退。

文字串是按順序匹配的,所以先找到有標記的節點一定是在文字串中靠前的,所以當我們遇到有標記節點時候,就說明找到一個單詞(題目保證單詞不會有包含關係),我們可以利用這個標記,我們插入的時候就可以令標記為串長,匹配成功時直接棧頂直接減個標記就行,再令位置成棧頂字元對應的位置繼續匹配下一個。

/*
Knowledge : Rubbish Algorithm
Work by :Gym_nastics
Time : O(AC)
*/
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int INF=0x3f3f3f3f;
const int Mod=1e9+7;
const int N=1e6+6;

int read() {
    int x=0,f=0;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) f|=(ch=='-');
    for(;isdigit(ch);ch=getchar()) x=(x<<1)+(x<<3)+(ch&15);
    return f?-x:x;
}

void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>9) print(x/10);
    putchar(x%10+48);
}

int Trie[N][27],e[N],fail[N],f[N],top,tot;
queue<int>q;char stc[N],S[N],t[N];

struct automaton{
    void Insert(char *s){
        int pos=0,len=strlen(s);
        for(int i=0;i<len;i++){
            int u=s[i]-'a'+1;
            if(!Trie[pos][u]) Trie[pos][u]=++tot;
            pos=Trie[pos][u];
        }e[pos]=len;
    }
    
    void Fail(){
        for(int i=1;i<=26;i++) if(Trie[0][i]) q.push(Trie[0][i]);
        while(!q.empty()){
            int u=q.front();q.pop();
            for(int i=1;i<=26;i++){
                if(Trie[u][i]) fail[Trie[u][i]]=Trie[fail[u]][i],q.push(Trie[u][i]);
                else Trie[u][i]=Trie[fail[u]][i];
            }
        }
    }
    
    void Query(char *s){
        int pos=0,len=strlen(s);
        for(int i=0;i<len;i++){
            int u=s[i]-'a'+1;
            pos=Trie[pos][u];
            stc[++top]=s[i],f[top]=pos;
            if(e[pos]) top-=e[pos],pos=f[top];
        }
        for(int i=1;i<=top;i++) putchar(stc[i]);
    } 
}AC;

signed main() {
   scanf("%s",t);int n=read();
   for(int i=1;i<=n;i++) scanf("%s",S),AC.Insert(S);
   AC.Fail();AC.Query(t);
   return 0;
}