bzoj 2434: [Noi2011]阿狸的打字机
Description
阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机。打字机上只有28个按键,分别印有26个小写英文字母和'B'、'P'两个字母。
经阿狸研究发现,这个打字机是这样工作的:
l 输入小写字母,打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。
l 按一下印有'B'的按键,打字机凹槽中最后一个字母会消失。
l 按一下印有'P'的按键,打字机会在纸上打印出凹槽中现有的所有字母并换行,但凹槽中的字母不会消失。
例如,阿狸输入aPaPBbP,纸上被打印的字符如下:
a
aa
ab
我们把纸上打印出来的字符串从1开始顺序编号,一直到n。打字机有一个非常有趣的功能,在打字机中暗藏一个带数字的小键盘,在小键盘上输入两个数(x,y)(其中1≤x,y≤n),打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。
阿狸发现了这个功能以后很兴奋,他想写个程序完成同样的功能,你能帮助他么?
Input
输入的第一行包含一个字符串,按阿狸的输入顺序给出所有阿狸输入的字符。
第二行包含一个整数m,表示询问个数。
接下来m行描述所有由小键盘输入的询问。其中第i行包含两个整数x, y,表示第i个询问为(x, y)。
Output
输出m行,其中第i行包含一个整数,表示第i个询问的答案。
Sample Input
3
1 2
1 3
2 3
Sample Output
1
0
HINT
1<=N<=10^5
Source
这题乱搞的暴力分有很多。。。
首先建出AC自动机,最暴力的做法就是把y串在trie上的每个点往上跳fail,如果能跳到x的结尾则说明出现了依次;
这等价于建出fail树后,y串在trie上的点有多少个在x的子树中;
直接想法就是在fail树上链剖,把y串到根节点的路径全部加1,然后查询x的值,但这样比较慢;
标算是把路径修改单点求值 改为 单点修改子树查询,这样只要再在AC自动机上跑一边,然后把询问相同的用vector挂个链,碰到单词结尾查询一下子树和即可,可以用树状数组实现;
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #include<queue> #include<vector> using namespace std; const int N=200050; int trie[N][26],fa[N],fail[N],pos[N],len,tt,cnt; int head[N*26],nxt[N*26],to[N*26],tot,flag,dfn[N],ed[N],T,c[N],ans[N]; char a[N]; void lnk(int x,int y){ to[++tot]=y,nxt[tot]=head[x],head[x]=tot; } void insert(){ int x=0; for(int i=1;i<=len;i++){ if(a[i]=='P') pos[++tt]=x; else if(a[i]=='B') x=fa[x]; else{ if(!trie[x][a[i]-'a']) trie[x][a[i]-'a']=++cnt; fa[cnt]=x,x=trie[x][a[i]-'a']; } } } void build(){ queue<int> q; fail[0]=0; for(int i=0;i<26;i++){ if(trie[0][i]) q.push(trie[0][i]); } while(!q.empty()){ int u=q.front();q.pop(); for(int i=0;i<26;i++){ int x=trie[u][i],j=fail[u]; if(!x) continue; q.push(x); while(j&&!trie[j][i]) j=fail[j]; if(trie[j][i]) fail[x]=trie[j][i]; } } for(int i=1;i<=cnt;i++) lnk(fail[i],i); } void dfs(int x){ dfn[x]=++T; for(int i=head[x];i;i=nxt[i]) dfs(to[i]); ed[x]=T; } void update(int x,int val){ for(int i=x;i<=T;i+=i&-i) c[i]+=val; } int query(int x){ int sum=0; for(int i=x;i;i-=i&-i) sum+=c[i]; return sum; } struct data{ int x,id; }; vector<data> p[N]; void solve(){ int m;cin>>m; for(int i=1;i<=m;i++){ int x,y;scanf("%d%d",&x,&y); p[pos[y]].push_back((data){x,i}); } int x=0; for(int i=1;i<=len;i++){ if(a[i]=='P'){ for(int j=0;j<p[x].size();j++){ data y=p[x][j]; ans[y.id]=query(ed[pos[y.x]])-query(dfn[pos[y.x]]-1); } } else if(a[i]=='B') update(dfn[x],-1),x=fa[x]; else x=trie[x][a[i]-'a'],update(dfn[x],1); } for(int i=1;i<=m;i++) printf("%d\n",ans[i]); } int main(){ scanf("%s",a+1);len=strlen(a+1); insert();build();dfs(0);solve(); }