BZOJ 2434 [Noi2011] 阿狸的打字机 Fail树

2017.3.17补…

题目大意：造一台打字机(滑稽),要求支持：1.在末尾添加一个字符 2.删除结尾字符 3.打印当前字符串 4.回答打印出的第x个字符串在第y个字符串中出现了几次。

询问一个字符串在另一个字符串中的出现次数可以想到利用AC自动机中构造出的Fail树的性质(类似BZOJ3172)。要如何回答询问呢？

查询x在y中出现了几次，就是查询Fail树上x的子树中有多少个结点属于y.

这样的话，维护一个dfs序的树状数组(因为想求子树和，且子树在dfs序中是连续的)，而对于一个单词，将他的所有节点插入到树状数组后就可以处理所有与其有关的询问。把所有询问建成一个邻接表，遇到P就处理所有有关这个字符串的询问，最后一起输出即可。

#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#define pb push_back
#define N 100005
using namespace std;
struct Edge {
int u,v,nxt,ans;
}query
;
int fir
,st
,ed
;
struct Node {
Node *ch[26],*fail,*pa;
int val,pos;
vector<Node*> to;
Node():val(0),fail(NULL),pa(NULL),pos(0) {
to.clear();
memset(ch,0,sizeof ch);
}
}*root=new Node();
int len,tot;
char s
;
void Insert() {
Node* o=root;
for(int i=0;i<len;i++) {
if(s[i]=='P') o->val=++tot;
else if(s[i]=='B') o=o->pa;
else {
int z=s[i]-'a';
if(!o->ch[z]) o->ch[z]=new Node(), o->ch[z]->pa=o;
o=o->ch[z];
}
}
return ;
}
void getFail() {
queue<Node*> q;
Node* o=root;
for(int i=0;i<26;i++)
if(o->ch[i]) {
q.push(o->ch[i]);
o->ch[i]->fail=o;
o->to.pb(o->ch[i]);
}
else o->ch[i]=o;
while(!q.empty()) {
o=q.front(); q.pop();
for(int i=0;i<26;i++)
if(o->ch[i]) {
q.push(o->ch[i]);
o->ch[i]->fail=o->fail->ch[i];
o->fail->ch[i]->to.pb(o->ch[i]);
}
else o->ch[i]=o->fail->ch[i];
}
return ;
}
int T;
namespace BIT {
int c
;
inline int lowbit(int x) {return x&-x;}
void add(int x,int y) {
for(int i=x;i<=T;i+=lowbit(i)) c[i]+=y;
return ;
}
int sum(int x) {
int tmp=0;
for(int i=x;i;i-=lowbit(i)) tmp+=c[i];
return tmp;
}
}
///build FailTree

void dfs(Node* o) {
o->pos=++T;
if(o->val) st[o->val]=T;
for(int i=0;i<o->to.size();i++)
dfs(o->to[i]);
if(o->val) ed[o->val]=T;
return ;
}
void solve() {
using namespace BIT;
Node* o=root;
for(int i=0;i<len;i++) {

if(s[i]=='B') add(o->pos,-1), o=o->pa;
else if(s[i]=='P')
for(int j=fir[o->val];j;j=query[j].nxt)
query[j].ans=sum(ed[query[j].v])-sum(st[query[j].v]-1);
else {
int z=s[i]-'a';
o=o->ch[z];
add(o->pos,1);
}
}
return ;
}
int main() {
scanf("%s",s);
len=strlen(s);
Insert();
getFail();
dfs(root);
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) {
int x,y;
scanf("%d%d",&x,&y);
query[i].u=y; query[i].v=x;
query[i].nxt=fir[y]; fir[y]=i;
}
solve();
for(int i=1;i<=n;i++) printf("%d\n",query[i].ans);
return 0;
}