[BZOJ]2434: [Noi2011]阿狸的打字机 AC自动机+树状数组

Description

阿狸喜欢收藏各种稀奇古怪的东西，最近他淘到一台老式的打字机。打字机上只有28个按键，分别印有26个小写英文字母和’B’、’P’两个字母。

经阿狸研究发现，这个打字机是这样工作的：

l 输入小写字母，打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。

l 按一下印有’B’的按键，打字机凹槽中最后一个字母会消失。

l 按一下印有’P’的按键，打字机会在纸上打印出凹槽中现有的所有字母并换行，但凹槽中的字母不会消失。

例如，阿狸输入aPaPBbP，纸上被打印的字符如下：

a

aa

ab

我们把纸上打印出来的字符串从1开始顺序编号，一直到n。打字机有一个非常有趣的功能，在打字机中暗藏一个带数字的小键盘，在小键盘上输入两个数(x,y)（其中1≤x,y≤n），打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。

阿狸发现了这个功能以后很兴奋，他想写个程序完成同样的功能，你能帮助他么？

题解：

先建AC自动机，要求x在y中出现了多少次，只要看root到y的路径上有多少个点可以通过不停地跳到fail，最终跳到x上，我们可以这样做：建fail树，顾名思义就是对于每一个点i，fail[i]到i建一条边，那么只要在root到y路径上每个点+1，最后用dfs序求一下x为根的子树和就好了。膜了之后1A感觉还是很爽的。

题解：

#include<cstdio>#include<cstring>#include<queue>#include<iostream>#include<algorithm>using namespace std;const int maxn=100010;int n;struct Trie{int fail,fa,son[28];}tr[maxn*10];struct Edge{int y,next;}e[maxn*10],e1[maxn];int len=0,last[maxn*10],len1=0,last1[maxn];int ID[maxn];void ins1(int x,int y,int i){    int t=++len1;    ID[t]=i;    e1[t].y=y;e1[t].next=last1[x];last1[x]=t;}void ins(int x,int y){    int t=++len;    e[t].y=y;e[t].next=last[x];last[x]=t;}void Clear(int x){    tr[x].fail=0;    for(int i=0;i<26;i++)tr[x].son[i]=-1;}char str[maxn];int z=0,pos[maxn],lp=0;int s[maxn*10];void add(int x,int y){for(int i=x;i<=z;i+=(i&-i))s[i]+=y;}int getsum(int x){int re=0;for(int i=x;i;i-=(i&-i))re+=s[i];return re;}void Build_Trie(){    int len=strlen(str),now=0;    for(int i=0;i<len;i++)    {        if(str[i]=='P')pos[++lp]=now;        else if(str[i]=='B')now=tr[now].fa;        else        {            int x=str[i]-'a';            if(tr[now].son[x]==-1)            tr[now].son[x]=++z,tr[z].fa=now,Clear(z);            now=tr[now].son[x];        }    }}queue<int>q;void work(){    q.push(0);    while(!q.empty())    {        int x=q.front();q.pop();        for(int i=0;i<26;i++)        {            int Son=tr[x].son[i],Fail=tr[x].fail;            if(Son>=0)            {                if(!x)tr[Son].fail=0;                else tr[Son].fail=tr[Fail].son[i];                q.push(Son);            }            else if(!x)tr[x].son[i]=0;            else tr[x].son[i]=tr[Fail].son[i];        }    }}int id=0,in[maxn*10],out[maxn*10];void dfs(int x){    in[x]=++id;    for(int i=last[x];i!=-1;i=e[i].next)    dfs(e[i].y);    out[x]=id;}int ans[maxn];void solve(){    int Y=0,len=strlen(str),now=0;    for(int i=0;i<len;i++)    {        if(str[i]=='P')        {            Y++;            for(int j=last1[Y];j;j=e1[j].next)            {                int X=e1[j].y,Pos=pos[X];                ans[ID[j]]=getsum(out[Pos])-getsum(in[Pos]-1);            }        }        else if(str[i]=='B')add(in[now],-1),now=tr[now].fa;        else now=tr[now].son[str[i]-'a'],add(in[now],1);    }}int main(){    memset(last,-1,sizeof(last));    Clear(0);tr[0].fa=0;    scanf("%s%d",str,&n);    Build_Trie();    work();    for(int i=1;i<=z;i++)ins(tr[i].fail,i);    for(int i=1;i<=n;i++)    {        int x,y;        scanf("%d%d",&x,&y);        ins1(y,x,i);    }    dfs(0);    z++;solve();    for(int i=1;i<=n;i++)    printf("%d\n",ans[i]);}