题目描述
打字机上只有28个按键,分别印有26个小写英文字母和'B'、'P'两个字母。经阿狸研究发现,这个打字机是这样工作的:
·输入小写字母,打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。
·按一下印有'B'的按键,打字机凹槽中最后一个字母会消失。
·按一下印有'P'的按键,打字机会在纸上打印出凹槽中现有的所有字母并换行,但凹槽中的字母不会消失。
例如,阿狸输入aPaPBbP,纸上被打印的字符如下:
a aa ab 我们把纸上打印出来的字符串从1开始顺序编号,一直到n。打字机有一个非常有趣的功能,在打字机中暗藏一个带数字的小键盘,在小键盘上输入两个数(x,y)(其中1≤x,y≤n),打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。
阿狸发现了这个功能以后很兴奋,他想写个程序完成同样的功能,你能帮助他么?
分析
看完题目确实懵了,不知道从哪里下手。和上一题相比,这道题考的是你对AC自动机理解的程度。
1.先说建立trie树的过程吧。他的操作序列是有规律的,每次会添加或删去一个字符,在某个时刻会输出当前字符代表这是一个新的单词。之前我们写的AC自动机一般是给了你n个单词,让你与一个长文本串匹配,这次其实一样。对于添加操作,那就从tire当前节点继续向下加点,如果是删去一个字符,那么我们向上走到now的父亲。
2.题目需要你求的是,第x个单词在第y个单词中出现了多少次。仔细想一下,AC自动机的fail指针指向的是什么,它指向的是和当前节点的字符串拥有最长相同后缀的字符串对应的节点。并且字符串一直向上爬,你会发现字母是一一对应的,否则它们之间不会有fail指针(具体情况画图试试,下面给一张例图)
进而,我们发现,如果y串中的某一个节点的fail指针指向了x串的末尾节点,那么代表x串在y串中出现了一次。因此,我们想知道y串中x串出现了多少次,只需要知道y串中有多少个fail指针指向x串的末尾节点。得到一个简单算法,每组询问,对y的节点遍历一次求一边答案,但是这个复杂度最大也是O(ML),依然过不了此题。
3.我们需要问题转化:x能对哪些y串造成影响呢?如果y中包含x,y一定有一条fail指针指向x,想到了什么?反向建图,我们可以得到一颗fail树。显然我们可以把数据离线处理,对q按照y的大小排序,每次到达某一个y的末尾节点,我们只需要查询它所对应的x末尾节点的子树的siz就是答案了。为了维护对子树siz的动态修改,我们需要dfs序。然后logn修改这道题就做完了。
感谢
点我跳转
#include<bits/stdc++.h> using namespace std; const int N=1e5+10; struct node{int x,y,id;}ask[N]; struct edge{int y,Next;}e[N]; char s[N]; int n,m,tot,cnt,tc,dfn,tr[N][30],fail[N],ans[N],l[N],r[N],lin[N],c[N],pos[N],fa[N]; void add(int x,int y){e[++tc].y=y;e[tc].Next=lin[x];lin[x]=tc;} void ad(int x,int val){for(x;x<=dfn;x+=x&(-x)) c[x]+=val;} int sum(int x){int res=0;for(;x;x-=x&(-x)) res+=c[x]; return res;} bool cmp(node a,node b){return a.y<b.y;} void build(){ int len=strlen(s+1),now=0; for(int i=1;i<=len;++i){ if(s[i]=='P') pos[++cnt]=now; else if(s[i]=='B') now=fa[now]; else{ int ch=s[i]-'a'; if(!tr[now][ch]) tr[now][ch]=++tot; fa[tr[now][ch]]=now,now=tr[now][ch]; } }n=cnt; queue<int>q; for(int i=0;i<26;++i){ if(tr[0][i]) fail[tr[0][i]]=0,q.push(tr[0][i]); } while(q.size()){ int x=q.front();q.pop(); for(int i=0;i<26;++i){ if(tr[x][i]){ int v=tr[x][i]; fail[v]=tr[fail[x]][i]; q.push(v); }else tr[x][i]=tr[fail[x]][i]; } } } void dfs(int x){ l[x]=++dfn; for(int i=lin[x];i;i=e[i].Next){ dfs(e[i].y); } r[x]=dfn; } void work(){ int len=strlen(s+1),now=0,k=1,cnt=0; for(int i=1;i<=len;++i){ if(s[i]=='P'){ cnt++; while(ask[k].y==cnt&&k<=m){ int tmp=pos[ask[k].x]; ans[ask[k].id]=sum(r[tmp])-sum(l[tmp]-1); k++; } }else if(s[i]=='B'){ ad(l[now],-1); now=fa[now]; }else{ int ch=s[i]-'a'; now=tr[now][ch]; ad(l[now],1); } } } int main(){ scanf("%s",s+1); scanf("%d",&m); for(int i=1;i<=m;++i) scanf("%d",ask[i].x,ask[i].y),ask[i].id=i; sort(ask+1,ask+m+1,cmp); build(); for(int i=1;i<=tot;++i) add(fail[i],i); dfs(0); work(); for(int i=1;i<=m;++i) printf("%d\n",ans[i]); return 0; }