【bzoj2434】阿狸的打字机 AC自动机+树状数组

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=2434 

【题解】

这是一道很神的题。

先建一个AC自动机，这里需要维护一下父结点，以便删除时返回上层。

然后构建一个fail树，得出dfs序，得出每个结点进出时间l[x]，r[x]，考虑这样一种暴力

对于一个询问x，y，查询自动机上root-y的每一个结点，沿着fail指针是否会走到x的结尾点

如果可以即答案+1

而在fail树中，变为查询自动机上root-y的所有结点中，有多少个在x的子树中？

只要在自动机上再重新走一遍，走到一个结点y，则将1-l[y]都+1，解决询问x，y（把y相同的链表串起来），即查询l[x]到r[x]的和。。。当遇到一个B时1-l[y]都-1

树状数组实现加减和区间求和

——转自hzwer

 

这个题解已经写得很好了，可惜我弱，写题时多次请教Cydiater大神，可见对AC自动机的理解不是很透彻，而且树状数组和dfs序什么的都是我不熟悉的。

这里我学到了AC自动机的进化版——trie图，就是构造fail指针时直接把子节点为空的点连到fail上去。

同时也初步接触到了fail树这种高端的东西。

收益匪浅，同时sro_Cydiater_orz     ←深入理解AC自动机已久的大神。

#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define MAXN 1000010
struct node{int y,next;}e[MAXN],qe[MAXN];
int n,m,cnt,len,T,Link[MAXN],qLink[MAXN],pos[MAXN],f[MAXN],q[MAXN],fail[MAXN],l[MAXN],r[MAXN],ans[MAXN],c[MAXN],tr[MAXN][27];
char ch[MAXN];
inline int read()
{
    int x=0,f=1;  char ch=getchar();
    while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getchar();}
    while(isdigit(ch))  {x=x*10+ch-'0';  ch=getchar();}
    return x*f;
}
void insert(int x,int y)  {e[++len].next=Link[x];Link[x]=len;e[len].y=y;}
int lowbit(int x) {return x&(-x);}
void add(int x,int v) {for(int i=x;i<=T;i+=lowbit(i)) c[i]+=v;}
int get(int x) {int sum=0; for(int i=x;i;i-=lowbit(i)) sum+=c[i]; return sum;}
void Insert()
{
    int now=0,id=0;
    for(int i=1;i<=n;i++)
    {
        if(ch[i]=='P') pos[++id]=now;
        else if(ch[i]=='B') now=f[now];
        else 
        {
            if(!tr[now][ch[i]-'a'])  tr[now][ch[i]-'a']=++cnt,f[cnt]=now; 
            now=tr[now][ch[i]-'a']; 
        }
    }
}
void build()
{
    int head=0,tail=0;
    for(int i=0;i<26;i++)  if(tr[0][i])  q[++tail]=tr[0][i];
    while(++head<=tail)
    {
        int x=q[head];
        for(int i=0;i<26;i++)
        {
            if(!tr[x][i]) tr[x][i]=tr[fail[x]][i];
            else {fail[tr[x][i]]=tr[fail[x]][i];  q[++tail]=tr[x][i];}
        }
    }
    for(int i=1;i<=cnt;i++)  insert(fail[i],i);
}
void init()
{
    scanf("%s",ch+1);  n=strlen(ch+1);  Insert();  build();  m=read();
    for(int i=1;i<=m;i++){int x=read(),y=read();qe[i].next=qLink[y];qLink[y]=i;qe[i].y=x;}
}
void dfs(int x) {l[x]=++T;for(int i=Link[x];i;i=e[i].next)dfs(e[i].y);r[x]=++T;}
void solve()
{
    dfs(0);  int now=0,id=0;
    for(int i=1;i<=n;i++)
    {
        if(ch[i]=='P')
        {
            id++;
            for(int j=qLink[id];j;j=qe[j].next)
            {
                int y=pos[qe[j].y];
                ans[j]=get(r[y])-get(l[y]-1);
            }
        }
        else if(ch[i]=='B')  add(l[now],-1),now=f[now];
        else now=tr[now][ch[i]-'a'],add(l[now],1);
    }
    for(int i=1;i<=m;i++)  printf("%d\n",ans[i]);
}
int main()
{
    //freopen("cin.in","r",stdin);
    //freopen("cout.out","w",stdout);
    init();
    solve();
    return 0;
}