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

题意

给你一些字符串。\(m\)次询问，每一次询问第\(x\)个字符串在\(y\)字符串中出现了多少次。（输入总长$ \le 10^5$, \(M \le 10^5\)）

分析

在ac自动机上，\(x\)字符串出现的所有位置就是其它节点的fail树上有这个节点的节点。即fail树中，\(x\)字符串终止节点的子树。

题解

根据分析，我们只要构造ac自动机和fail树，按dfs序依次便历ac自动机，将节点到根的路径打上标记。每到一个终止节点，就更新以这个节点为模板的匹配串的答案，即这些节点子树的和。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
char buf[20*N], *is=buf;
int ihead[N], cnt, last[N], c[N][26], fail[N], pos[N], FF[N], LL[N], tot, su[N], ans[N];
inline bool isdig(const char &c) {
    return c>='0'&&c<='9';
}
inline bool isok(const char &c) {
    return (c>='a'&&c<='z')||c=='P'||c=='B';
}
inline int getint() {
    register int x=0;
    for(; !isdig(*is); ++is);
    for(; isdig(*is); ++is) {
        x=x*10+*is-48;
    }
    return x;
}
inline void putint(int x) {
    if(x==0) {
        *is++='0';
    }
    else {
        static int s[10], top;
        for(top=0; x; x/=10) s[++top]=x%10;
        while(top) {
            *is++=s[top--]+'0';
        }
    }
    *is++='\n';
}
struct E {
    int next, to, id;
}e[N];
inline void add(int x, int y, int id=-1) {
    e[++cnt]=(E){ihead[x], y, id}; ihead[x]=cnt;
}
inline void adds(int x, int s) {
    for(; x<=tot; x+=x&-x) {
        su[x]+=s;
    }
}
inline int sum(int x) {
    int y=0;
    for(; x; x-=x&-x) {
        y+=su[x];
    }
    return y;
}
inline void bfs() {
    static int q[N], fr, ta;
    fr=ta=0;
    q[ta++]=0;
    while(fr!=ta) {
        int x=q[fr++];
        for(int ch=0; ch<26; ++ch) {
            if(c[x][ch]) {
                int y=c[x][ch];
                q[ta++]=y;
                if(x==0) {
                    add(0, y);
                    continue;
                }
                fail[y]=c[fail[x]][ch];
                add(fail[y], y);
            }
            else {
                c[x][ch]=c[fail[x]][ch];
            }
        }
    }
}
inline void dfs(int x) {
    static int fid=0;
    FF[x]=++fid;
    for(int i=ihead[x]; i; i=e[i].next) {
        dfs(e[i].to);
    }
    LL[x]=fid;
}
inline void getans(int x) {
    adds(FF[x], 1);
    for(int i=ihead[x]; i; i=e[i].next) {
        int y=e[i].to;
        ans[e[i].id]=sum(LL[y])-sum(FF[y]-1);
    }
    for(int ch=0; ch<26; ++ch) {
        if(c[x][ch] && (x==0 || c[x][ch]!=c[fail[x]][ch])) {
            getans(c[x][ch]);
        }
    }
    adds(FF[x], -1);
}
void init() {
    fread(buf, 1, sizeof buf, stdin);
    for(; !isok(*is); ++is);
    int now=0, n=0, len=0;
    for(; isok(*is); ++is) {
        if(*is=='P') {
            pos[++n]=now;
        }
        else if(*is=='B') {
            now=last[len--];
        }
        else {
            int ch=*is-'a';
            if(!c[now][ch]) {
                c[now][ch]=++tot;
            }
            last[++len]=now;
            now=c[now][ch];
        }
    }
    bfs();
    dfs(0);
}
int main() {
    init();
    memset(ihead, 0, sizeof(int)*(tot+1));
    cnt=0;
    int m=getint();
    for(int i=0; i<m; ++i) {
        int x, y;
        x=getint();
        y=getint();
        add(pos[y], pos[x], i);
    }
    ++tot;
    getans(0);
    is=buf;
    for(int i=0; i<m; ++i) {
        putint(ans[i]);
    }
    fwrite(buf, 1, sizeof(char)*(is-buf), stdout);
    return 0;
}

转载于:https://www.cnblogs.com/iwtwiioi/p/4985740.html