【刷题】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

aPaPBbP

3

1 2

1 3

2 3

Sample Output

2

1

0

HINT

1<=N<=10^5

1<=M<=10^5

输入总长<=10^5

Solution

AC自动机

不知道性质完全做不了

如果AC自动机中的两个串 \(p\) 和 \(s\) ，满足 \(p\) 在 \(s\) 中出现了 \(k\) 次，那么 \(k\) 等于 \(p\) 串结束点在fail树上的子树中有多少个节点属于串 \(s\)

维护树中一个节点的子树和，就随便搞了，我用的BIT

然后就可以离线了

首先把串按照题目含义建出AC自动机

然后getfail

对询问排序

开始算答案，重新跟着给的串走一遍，达到一个点，把它的贡献加入BIT，离开一个点则减去，那么BIT中保存的就一定是当前达到的点到根路径上的点的贡献

（注意，加贡献一定是加在这个点在fail树中的位置，而不是AC自动机上的位置）

所以如果当前到达了一个询问，就可以直接在BIT查答案了

#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
const int MAXN=100000+10,MAXM=100000+10;
int ch[MAXN][30],fail[MAXN],n,m,cnt,st[MAXN],e,beg[MAXN],nex[MAXN<<1],to[MAXN<<1],nt,ans[MAXN],ed[MAXN],fa[MAXN],fl[MAXN],ps[MAXN];
char s[MAXN];
std::queue<int> q;
struct node{
	int x,y,id;
	inline bool operator < (const node &A) const {
		return y<A.y;
	};
};
node qs[MAXM];
struct BIT{
	int C[MAXN];
	inline int lowbit(int x)
	{
		return x&(-x);
	}
	inline void add(int x,int k)
	{
		while(x<=nt)C[x]+=k,x+=lowbit(x);
	}
	inline int sum(int x)
	{
		int res=0;
		while(x>0)res+=C[x],x-=lowbit(x);
		return res;
	}
};
BIT T;
template<typename T> inline void read(T &x)
{
	T data=0,w=1;
	char ch=0;
	while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
	if(ch=='-')w=-1,ch=getchar();
	while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
	x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
	if(ch!='\0')putchar(ch);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void insert(int x,int y)
{
	to[++e]=y;
	nex[e]=beg[x];
	beg[x]=e;
}
inline void init()
{
	int x=0;
	for(register int i=0,lt=strlen(s);i<lt;++i)
		if(s[i]=='P')fl[x]=++n,ps[n]=x;
		else if(s[i]=='B')x=fa[x];
		else if(!ch[x][s[i]-'a'+1])fa[cnt+1]=x,x=ch[x][s[i]-'a'+1]=++cnt;
		else x=ch[x][s[i]-'a'+1];
}
inline void getfail()
{
	for(register int i=1;i<=26;++i)
		if(ch[0][i])q.push(ch[0][i]);
	while(!q.empty())
	{
		int x=q.front();
		q.pop();
		for(register int i=1,y,z;i<=26;++i)
			if(ch[x][i])
			{
				y=ch[x][i],z=fail[x];
				while(z&&!ch[z][i])z=fail[z];
				fail[y]=ch[z][i];
				q.push(y);
			}
	}
}
inline void dfs(int x,int f)
{
	st[x]=++nt;
	for(register int i=beg[x];i;i=nex[i])
		if(to[i]==f)continue;
		else dfs(to[i],x);
	ed[x]=nt;
}
inline void solve()
{
	int x=0,nq=1;
	for(register int i=0,lt=strlen(s);i<lt;++i)
	{
		if(s[i]=='P')
			while(fl[x]==qs[nq].y)ans[qs[nq].id]=T.sum(ed[ps[qs[nq].x]])-T.sum(st[ps[qs[nq].x]]-1),++nq;
		else if(s[i]=='B')T.add(st[x],-1),x=fa[x];
		else x=ch[x][s[i]-'a'+1],T.add(st[x],1);
	}
}
int main()
{
	scanf("%s",s);
	init();getfail();
	for(register int i=1;i<=cnt;++i)insert(i,fail[i]),insert(fail[i],i);
	read(m);
	for(register int i=1;i<=m;++i)read(qs[i].x),read(qs[i].y),qs[i].id=i;
	dfs(0,-1);
	std::sort(qs+1,qs+m+1);
	solve();
	for(register int i=1;i<=m;++i)write(ans[i],'\n');
	return 0;
}