【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+;
 char s[N];
 struct edge{ int next,to; edge(int next=,int to=) : next(next),to(to){}; }e[N];
 struct trie{ int th; trie *next[],*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=,r=,df,n,c2;
 int first[N],st[N],ed[N],BIT[N*],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=,a2=;
     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!=) modify(st[pn->th],);
     return pn;
 }

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

 void build_fail(){
     trie *p=q[]=root;
     root->th=;
     for (int i=;i<;i++) if (pn) q[++r]=pn,pnf=root,pn->th=r,addedge(,pn->th);
     while (l<r){
         p=q[++l];
         for (int i=;i<;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=;i<=n;i++) {
         scanf("%d%d",&Q[i].x,&Q[i].y);
         Q[i].id=i;
     }
     sort(Q+,Q++n,oper);
     for (int l=strlen(s),i=;i<l;i++){
         if (s[i]>='a' && s[i]<='z') now=insert(now,s[i]-'a',);
         if (s[i]=='B') now=remove(now,);
         if (s[i]=='P') sth[++c2]=now;
     }
     build_fail();
     dfs();
     now=root;
     for (int i=,j=;i<=n;i++){
         while (sth[Q[i].y]!=now) {
             if (s[j]>='a' && s[j]<='z') now=insert(now,s[j]-'a',);
             if (s[j]=='B') now=remove(now,);
             j++;
         }
         ans[Q[i].id]=query(st[sth[Q[i].x]->th],ed[sth[Q[i].x]->th]);
     }
     for (int i=;i<=n;i++) printf("%d\n",ans[i]);
     return ;
 }

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