P3121 [USACO15FEB]Censoring G

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;
}
posted @ 2022-03-16 22:24  Gym_nastics  阅读(48)  评论(0)    收藏  举报