[BZOJ]3940: [Usaco2015 Feb]Censoring AC自动机

来源:互联网 发布:郑州软件开发工资 编辑:程序博客网 时间:2024/06/05 14:44

Description
FJ把杂志上所有的文章摘抄了下来并把它变成了一个长度不超过10^5的字符串S。他有一个包含n个单词的列表,列表里的n个单词
记为t_1…t_N。他希望从S中删除这些单词。
FJ每次在S中找到最早出现的列表中的单词(最早出现指该单词的开始位置最小),然后从S中删除这个单词。他重复这个操作直到S中
没有列表里的单词为止。注意删除一个单词后可能会导致S中出现另一个列表中的单词
FJ注意到列表中的单词不会出现一个单词是另一个单词子串的情况,这意味着每个列表中的单词在S中出现的开始位置是互不相同的
请帮助FJ完成这些操作并输出最后的S

题解:

跟BZOJ3942差不多,只不过把KMP换成了AC自动机。3942题解。

代码:

#include<cstdio>#include<cstring>#include<queue>#include<iostream>#include<algorithm>using namespace std;const int maxn=100010;int n;struct Trie{int fail,son[28],dep;bool mark;}tr[maxn*10];void Clear(int x){    tr[x].fail=0;tr[x].mark=false;    for(int i=0;i<26;i++)tr[x].son[i]=-1;}char str[maxn],s[maxn],ans[maxn];int z=0,pp[maxn];void Build_Trie(){    int len=strlen(str),now=0;    for(int i=0;i<len;i++)    {        int x=str[i]-'a';        if(tr[now].son[x]==-1)        tr[now].son[x]=++z,tr[z].dep=tr[now].dep+1,Clear(z);        now=tr[now].son[x];    }    tr[now].mark=true;}queue<int>q;void work(){    q.push(0);    while(!q.empty())    {        int x=q.front();q.pop();        for(int i=0;i<26;i++)        {            int Son=tr[x].son[i],Fail=tr[x].fail;            if(Son>=0)            {                if(!x)tr[Son].fail=0;                else tr[Son].fail=tr[Fail].son[i];                q.push(Son);            }            else if(!x)tr[x].son[i]=0;            else tr[x].son[i]=tr[Fail].son[i];        }    }}int main(){    Clear(0);tr[0].dep=0;    scanf("%s%d",s,&n);    for(int i=1;i<=n;i++)    {        scanf("%s",str);        Build_Trie();    }    work();    int p=0,ls=strlen(s),len=0;    for(int i=0;i<ls;i++)    {        int x=s[i]-'a';        while(p&&!tr[p].son[x])p=tr[p].fail;        if(tr[p].son[x])p=tr[p].son[x];        ans[++len]=s[i];pp[len]=p;        if(tr[p].mark)        {            len-=tr[p].dep;            p=pp[len];        }    }    for(int i=1;i<=len;i++)printf("%c",ans[i]);}