BZOJ2434: [NOI2011]阿狸的打字机（洛谷P2414）

AC自动机 树状数组

BZOJ题目传送门
洛谷题目传送门

可以发现整个操作过程就是建立一颗Trie树。而查询第x个打印的字符串在第y个打印的字符串中出现的次数其实就是查询有多少个y的节点的fail指针指向在x的结束节点。

我们把fail树建起来，把y都打上1，就会发现询问变成在x结束节点所在的子树和。那么dfs序之后用树状数组维护一下就好了。

代码：

#include<queue>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define F inline
#define V F void
using namespace std;
struct edge{ int next,to; }ed[N<<1];
struct query{ int next,to,id; }qd[N<<1];
int m,k,nd,ti,num,to[N][26],nxt[N],fa[N];
int t[N],id[N],L[N],R[N],ans[N],h[N],qh[N],Id[N];
char s[N]; queue <int> q;
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
V sread(char *s){
    char ch=readc(); int p=1;
    while (!islower(ch)&&!isupper(ch)) ch=readc();
    while (islower(ch)||isupper(ch)) s[p++]=ch,ch=readc();
}
V writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
V _write(int x){ writec(x),putchar('\n'); }
V addedge(int x,int y) { ed[++k]=(edge){h[x],y},h[x]=k; }
V addedge(int x,int y,int z) { qd[++k]=(query){qh[x],y,z},qh[x]=k; }
V build(){
    for (int i=1,x=0,c;s[i];i++)
        if (islower(s[i])){
            if (!to[x][c=s[i]-'a']) to[x][c]=++nd;
            fa[to[x][c]]=x,x=to[x][c];
        }
        else if (s[i]=='B') x=fa[x];
        else id[++num]=x,Id[x]=num;
    while (!q.empty()) q.pop();
    for (int i=0,v;i<26;i++) if (v=to[0][i]) q.push(v),addedge(0,v);
    while (!q.empty()){
        int x=q.front(),now=nxt[x]; q.pop();
        for (int i=0,v;i<26;i++)
            if (v=to[x][i])
                q.push(v),nxt[v]=to[now][i],addedge(nxt[v],v);
            else to[x][i]=to[now][i];
    }
}
void dfs(int x){
    L[x]=++ti; for (int i=h[x];i;i=ed[i].next) dfs(ed[i].to); R[x]=ti;
}
#define t(x) (x&(-x))
V nsrt(int x,int w){ for (;x<=ti;x+=t(x)) t[x]+=w; }
F int srch(int x){ int i=0; for (;x;x-=t(x)) i+=t[x]; return i; }
int main(){
    sread(s),m=_read(),build(),dfs(0),k=0;
    for (int i=1,x,y,z;i<=m;i++) x=_read(),y=_read(),addedge(y,x,i);
    for (int i=1,x=0,c;s[i];i++)
        if (islower(s[i])) x=to[x][c=s[i]-'a'],nsrt(L[x],1);
        else if (s[i]=='B') nsrt(L[x],-1),x=fa[x];
        else
            for (int j=qh[Id[x]],v;j;j=qd[j].next)
                v=qd[j].to,ans[qd[j].id]=srch(R[id[v]])-srch(L[id[v]]-1);
    for (int i=1;i<=m;i++) _write(ans[i]);
    return 0;
}