【BZOJ3940】[USACO2015 Feb] Censoring （AC自动机的小应用）

最新推荐文章于 2020-07-18 07:50:13 发布

chenxiaoran666

最新推荐文章于 2020-07-18 07:50:13 发布

阅读量230

点赞数

分类专栏： BZOJ AC自动机文章标签： AC自动机

本文链接：https://blog.csdn.net/chenxiaoran666/article/details/81876322

版权

BZOJ 同时被 2 个专栏收录

35 篇文章 0 订阅

订阅专栏

AC自动机

6 篇文章 0 订阅

订阅专栏

点此看题面
大致题意： 给你一个文本串和 $N$ 个模式串，要你将每一个模式串从文本串中删去。（此题是【bzoj3942】[Usaco2015 Feb]Censoring的升级版）

$A C$ 自动机

这是一道 $A C$ 自动机的简单运用题。

$L i n k$

$A C$ 自动机详见博客初学AC自动机

题解

我们可以用一个 $a n s [$ $]$ 字符数组来存储最后的答案，并用一个 $l s t [$ $]$ 数组来记录字符串中每一位上次被访问时所到达的 $T r i e$ 上的节点。

如果找到了一个长度为 $l e n$ 的模式串要删去，我们可以将 $a n s$ 的长度 $len_{ans}$ 减去 $l e n$ ，然后更新当前在 $T r i e$ 上访问到的节点为 $lst[len_{ans}]$ 。

这样不断操作，就可以求出答案了。

代码

#include<bits/stdc++.h>
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
#define LL long long
#define ull unsigned long long
#define swap(x,y) (x^=y,y^=x,x^=y)
#define tc() (A==B&&(B=(A=ff)+fread(ff,1,100000,stdin),A==B)?EOF:*A++)
#define SUM 100000
char ff[100000],*A=ff,*B=ff;
using namespace std;
int n,rt=1,tot=1,lst[SUM+5];
string st;
char ans[SUM+5];
struct Trie
{
    int Son[26],Next,Len;
}node[SUM+5];
queue<int> q;
inline void read(int &x)
{
    x=0;static char ch;
    while(!isdigit(ch=tc()));
    while(x=(x<<3)+(x<<1)+ch-48,isdigit(ch=tc()));
}
inline void read_string(string &x)
{
    x="";static char ch;
    while(isspace(ch=tc()));
    while(x+=ch,!isspace(ch=tc())) if(!(~ch)) return;
}
inline void Insert(string st)
{
    register int i,nxt,x=rt,len=st.length();
    for(i=0;i<len;++i)
    {
        if(!node[x].Son[nxt=st[i]-97]) node[x].Son[nxt]=++tot;
        x=node[x].Son[nxt];
    }
    node[x].Len=len;//在当前模式串最后到达的节点记录下当前模式串的长度
}
inline void GetNext()
{
    register int i,k;q.push(rt);
    while(!q.empty())
    {
        k=q.front(),q.pop();
        for(i=0;i<26;++i)
        {
            if(k^rt)
            {
                if(!node[k].Son[i]) node[k].Son[i]=node[node[k].Next].Son[i];
                else node[node[k].Son[i]].Next=node[node[k].Next].Son[i],q.push(node[k].Son[i]);
            }
            else
            {
                if(!node[k].Son[i]) node[k].Son[i]=rt;
                else node[node[k].Son[i]].Next=rt,q.push(node[k].Son[i]);
            }
        }
    }
}
inline void AC_Automation()
{
    register int i,j,x=rt,len=st.length(),len_ans=0;
    for(GetNext(),i=0;i<len;++i)
    {
        if(!(x=node[x].Son[(ans[++len_ans]=st[i])-97])) {lst[len_ans]=x=rt;continue;}
        lst[len_ans]=x;//用lst[]数组存储下当前位置所对应的Trie上的节点
        if(node[x].Len) len_ans-=node[x].Len,x=lst[len_ans];//如果当前位置上有一个字符串，就将其删去，并更新当前Trie上的节点
    }
    ans[len_ans+1]=0;
}
int main()
{
    register int i,j;register string s;
    for(read_string(st),read(n),i=1;i<=n;++i) read_string(s),Insert(s);
    return AC_Automation(),puts(ans+1),0;
}