【bzoj2434】: [Noi2011]阿狸的打字机 字符串-AC自动机-BIT

【bzoj2434】: [Noi2011]阿狸的打字机

x串在y串上的匹配次数就是y在自动机所有节点上能够通过fail走到x最后一个节点的个数

（就是y串任意一个前缀的后缀能匹配到x的个数）和【bzoj3172】: [Tjoi2013]单词差不多

把fail指针反向就是x的子树的和 用dfs序+BIT就可以维护了

如果把y串一个个走自动机显然会TLE

但是如果按照打字顺序模拟'a'-'z' 在trie上前进一格单点+1 'B'在trie上后退一格单点-1

那么显然可以得到所有y串但只需要O(n)

然后离线处理一下就好了

/* http://www.cnblogs.com/karl07/ */
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int N=1e5+5;
char s[N];
struct edge{ int next,to; edge(int next=0,int to=0) : next(next),to(to){}; }e[N];
struct trie{ int th; trie *next[26],*fail,*fa; }t[N],*NEW=t,*root=t,*q[N],*now=t,*sth[N];
struct Query{int x,y,id;}Q[N];
int ade,l=1,r=1,df,n,c2;
int first[N],st[N],ed[N],BIT[N*2],ans[N];
#define pn p->next[i]
#define pnf p->next[i]->fail
#define lowbit(x) (x&-x)

void modify(int x,int m){
    for (;x<=N;x+=lowbit(x)) BIT[x]+=m; 
}

int query(int l,int r){
    int a1=0,a2=0;
    for (;r;r-=lowbit(r)) a1+=BIT[r];
    for (l--;l;l-=lowbit(l)) a2+=BIT[l];
    return a1-a2;
}

void addedge(int x,int y){
    e[++ade]=edge(first[x],y),first[x]=ade;
}

trie *new1(trie *p){ NEW++; NEW->fa=p; return NEW; }

trie *insert(trie *p,int i,int x){
    if (!pn) pn=new1(p);
    if (x!=0) modify(st[pn->th],1);
    return pn;
}

trie *remove(trie *p,int x){
    if (x!=0) modify(st[p->th],-1);
    return p->fa;
}

void build_fail(){
    trie *p=q[1]=root;
    root->th=1;
    for (int i=0;i<26;i++) if (pn) q[++r]=pn,pnf=root,pn->th=r,addedge(1,pn->th);
    while (l<r){
        p=q[++l];
        for (int i=0;i<26;i++){
            if (pn){
                q[++r]=pn,pn->th=r;
                for (pnf=p->fail ; pnf!=root && !pnf->next[i] ; pnf=pnf->fail);
                if (pnf->next[i]) pnf=pnf->next[i];
                addedge(pnf->th,pn->th);
            }
        }
    }
}

void dfs(int p){
    st[p]=++df;
    for (int x=first[p];x;x=e[x].next){
        if (e[x].to) dfs(e[x].to);
    }
    ed[p]=df;
}

bool oper(Query a, Query b){return a.y<b.y;}

int main(){
    scanf("%s",s);
    scanf("%d",&n);
    for (int i=1;i<=n;i++) {
        scanf("%d%d",&Q[i].x,&Q[i].y);
        Q[i].id=i;
    }
    sort(Q+1,Q+1+n,oper);
    for (int l=strlen(s),i=0;i<l;i++){
        if (s[i]>='a' && s[i]<='z') now=insert(now,s[i]-'a',0);
        if (s[i]=='B') now=remove(now,0);
        if (s[i]=='P') sth[++c2]=now;
    }
    build_fail();
    dfs(1);
    now=root;
    for (int i=1,j=0;i<=n;i++){
        while (sth[Q[i].y]!=now) {
            if (s[j]>='a' && s[j]<='z') now=insert(now,s[j]-'a',1);        
            if (s[j]=='B') now=remove(now,1);
            j++;
        }
        ans[Q[i].id]=query(st[sth[Q[i].x]->th],ed[sth[Q[i].x]->th]);
    }
    for (int i=1;i<=n;i++) printf("%d
",ans[i]); 
    return 0;
}

竟然过了样例一次就A了。。