2434: [Noi2011]阿狸的打字机

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 2022  Solved: 1158
[Submit][Status][Discuss]

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

aPaPBbP
3
1 2
1 3
2 3

Sample Output

2
1
0

HINT

1<=N<=10^5

1<=M<=10^5
输入总长<=10^5

Source

Trie

Solution

好题,值得多次练习

AC自动机无疑,建出Fail指针的同时,根据Fail指针反向建树,即Fail树

以x为根的子树沿着fail指针倒着走能找到多少个y路径上的点就说明x在y上出现过几次。

那么利用DFS,求出每个点的时间戳(即每个节点最先到达的时间和最后到达时间),那么发现,对于子树,是一种类似前缀和的存在

那么用线段树或者树状数组来维护一下

值得注意的几点:(皆为亲身经历)

首先在建Trie树的过程中,应该考虑直接对整个操作串进行处理,不断分离去插入会TLE

对于操作串的长度,不应该直接在for循环中使用strlen(),应该先记录下来len=strlen()后循环中直接用len,否则更TLE

不要企图在线处理,应该优先考虑离线处理,否则也会出现奇怪的错误

对于Fail树,应该建单向边,并非双向边,否则可能出现错误

数组要开的够大,题目中10^5,开到10^5+100,Tyvj秒A,CodeVS秒A,BZOJ狂WA不止,怒开2×10^5秒A...不知所措

Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
int read()
{
int x=;char ch=getchar();
while(ch<''||ch>'')ch=getchar();
while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
return x;
}
#define N 210000
int son[N][],fail[N],q[N],sz,t,len,m,ans[N],fa[N],loc[N];
char s[N],opt[N];
struct data{int to,next;}edge[N],ask[N];
int cn,cnt,head[N],heada[N],lastq[N];
void addedge(int u,int v)
{cnt++;edge[cnt].to=v;edge[cnt].next=head[u];head[u]=cnt;}
void addask(int x,int y,int z)
{ask[z].next=heada[y];heada[y]=z;ask[z].to=x;}
int tree[N];
inline int lowbit(int x){return x&(-x);}
void change(int loc,int val)
{
for (int i=loc; i<=t; i+=lowbit(i))
tree[i]+=val;
}
int query(int loc)
{
int ans=;
for (int i=loc; i>=; i-=lowbit(i))
ans+=tree[i];
return ans;
}
int L[N],R[N];
void dfs(int x)
{
L[x]=++t;
for (int i=head[x]; i; i=edge[i].next)
dfs(edge[i].to);
R[x]=++t;
//printf("L=%d R=%d\n",L[x],R[x]);
}
void clear()
{sz=;for (int i=; i<; i++) son[][i]=;}
void buildTrie()
{
int x=,id=;
for (int i=; i<len; i++)
if (opt[i]=='P') loc[++id]=x;
else if (opt[i]=='B') x=fa[x];
else
{
if (!son[x][opt[i]-'a'])
son[x][opt[i]-'a']=++sz,fa[sz]=x;
x=son[x][opt[i]-'a'];
}
}
void buildFail()
{
int he=,ta=; q[]=; fail[]=;
while (he<ta)
{
int now=q[he++];
for (int i=; i<; i++)
{
if (!son[now][i]) continue;
int ff=fail[now];
while (!son[ff][i]) ff=fail[ff];
fail[son[now][i]]=son[ff][i];
q[ta++]=son[now][i];
}
}
for (int i=; i<=sz; i++)
addedge(fail[i],i);
}
void work()
{
int now=,id=;
change(L[],);
for (int i=; i<len; i++)
if (opt[i]=='P')
{
id++;
for (int j=heada[id]; j; j=ask[j].next)
ans[j]=query(R[loc[ask[j].to]])-query(L[loc[ask[j].to]]-);
}
else if (opt[i]=='B') change(L[now],-),now=fa[now];
else now=son[now][opt[i]-'a'],change(L[now],);
}
//ACMachine
int main()
{
clear();
scanf("%s",opt); len=strlen(opt);
buildTrie(); buildFail();
m=read();
for (int x,y,i=; i<=m; i++)
x=read(),y=read(),addask(x,y,i);
dfs(); work();
for (int i=; i<=m; i++)
printf("%d\n",ans[i]);
return ;
}

BZOJ数组大小被坑艰辛史

【BZOJ-2434】阿狸的打字机    AC自动机 + Fail树 + DFS序 + 树状数组-LMLPHP

05-08 15:11