首页 > 代码库 > bzoj 2434: [Noi2011]阿狸的打字机
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
Trie
这题乱搞的暴力分有很多。。。
首先建出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();
}
bzoj 2434: [Noi2011]阿狸的打字机