阿狸的打字机 AC自动机 FAIL树 树状数组

http://hzwer.com/5423.html
http://www.cnblogs.com/Tunix/p/4559733.html

#include <cstdio>
#include <string>
#include <cstring>
#include <queue>
#include <algorithm>
#include <functional>
#include <vector>
#include <iomanip>
#include <cmath>
#include <iostream>
#include <sstream>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
int n,m,cnt;
int T,l[100005],r[100005],t[200005];
int last[100005],lastq[100005],pos[100005],ans[100005];
char ch[100005];

struct fail_tree{
    int to,next,v;
}e[100005],que[100005];

void add(int x,int val)
{
    for(int i=x;i<=T;i+=i&-i)t[i]+=val;
}
int query(int x)
{
    int sum=0;
    for(int i=x;i;i-=i&-i)sum+=t[i];
    return sum;
}
void insert(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}
struct acm{
    int cnt;
    int fa[100005],fail[100005],next[100005][26];
    int q[100005];
    acm(){
        cnt=1;
        for(int i=0;i<26;i++)
            next[0][i]=1;
    }
    void build(){
        int now=1,id=0;
        for(int i=0;i<n;i++)
            if(ch[i]=='P')pos[++id]=now;
            else if(ch[i]=='B')now=fa[now];
            else
            {
                if(!next[now][ch[i]-'a'])
                {
                    next[now][ch[i]-'a']=++cnt;
                    fa[cnt]=now;
                }
                now=next[now][ch[i]-'a'];
            }
    }
    void buildfail(){
        int head=0,tail=1;
        fail[1]=0;q[0]=1;
        while(head!=tail)
        {
            int now=q[head];head++;
            for(int i=0;i<26;i++)
                if(next[now][i])
                {
                    int v=next[now][i];
                    int k=fail[now];
                    while(!next[k][i])k=fail[k];
                    fail[v]=next[k][i];
                    q[tail++]=v;
                }
        }
    }
    void solve(){
        int now=1,id=0;
        add(l[1],1);
        for(int i=0;i<n;i++)
            if(ch[i]=='P')
            {
                id++;
                for(int x=lastq[id];x;x=que[x].next)
                {
                    int t=pos[que[x].to];
                    ans[x]=query(r[t])-query(l[t]-1);
                }
            }
            else if(ch[i]=='B')add(l[now],-1),now=fa[now];
            else now=next[now][ch[i]-'a'],add(l[now],1);;
    }
}acm;
void dfs(int x)
{
    l[x]=++T;
    for(int i=last[x];i;i=e[i].next)
        dfs(e[i].to);
    r[x]=++T;
}
int main()
{
    scanf("%s",ch);
    n=strlen(ch);
    acm.build();
    acm.buildfail();
    for(int i=1;i<=acm.cnt;i++)
        insert(acm.fail[i],i);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        que[i].next=lastq[y];
        lastq[y]=i;
        que[i].to=x;
    }
    dfs(0);
    acm.solve();
    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
    return 0;
}