一、题目
二、解法
我们先对输入的字符串建出 A C AC AC自动机,自动机的点数是 O ( n ) O(n) O(n)的,建法如下:
- 遇到小写字母,跳自动机上的边,看是否需要新建点。
- 遇到
P
,存下字符串对应在自动机上的点编号。 - 遇到
B
,向当前点的父亲上跳。
考虑 x x x串在 y y y串的出现次数,具体过程就是在自动机上跳到 y y y,路径上的每一个点都去跳 f a i l fail fail,跳到 x x x对应点的次数即为出现次数。还有另一种理解方法,如果我们建出 f a i l fail fail树,把 y y y路径上的所有点在树上打标记,我们要做的就是查询 x x x为根的子树内有多少个被打上标记的点。
可以离线询问,这样就方便了打标记和撤回标记,至于查询子树,我们可以用dfn+树状数组
实现。我们逐个访问第一行字符串中的字符:
- 遇到小写字符,向下跳( t r i e trie trie树),打标记(树状数组单点修改)。
- 遇到
P
,回答离线下来的问题(树状数组区间查询)。 - 遇到
B
,向父亲跳( t r i e trie trie树),删除标记。
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),详见代码,自认为写的比较好看qwq。
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int M = 100005;
int read()
{
int x=0,flag=1;char c;
while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*flag;
}
int n,a[M],tot,f[M],c[M][26],val[M],fail[M],fa[M],cnt;
int m,len,Index,dfn[M],out[M],bit[M],ans[M];
char s[M];
struct edge
{
int v,next;
edge(int V=0,int N=0) : v(V) , next(N) {}
}e[2*M];
struct node
{
int x,id;
node(int X=0,int I=0) : x(X) , id(I) {}
};vector<node> g[M];
void link(int u,int v)
{
e[++tot]=edge(v,f[u]),f[u]=tot;
e[++tot]=edge(u,f[v]),f[v]=tot;
}
void ins()
{
scanf("%s",s);
len=strlen(s);
for(int i=0,now=0;i<len;i++)
{
if(s[i]>='a' && s[i]<='z')
{
int v=s[i]-'a';
if(!c[now][v]) c[now][v]=++cnt,fa[cnt]=now;
now=cnt;
}
if(s[i]=='P')
a[++n]=now;
if(s[i]=='B')
now=fa[now];
}
}
void dfs(int u,int p)
{
dfn[u]=++Index;
for(int i=f[u];i;i=e[i].next)
{
int v=e[i].v;
if(v^p) dfs(v,u);
}
out[u]=Index;
}
void build()
{
queue<int> q;
for(int i=0;i<26;i++) if(c[0][i]) q.push(c[0][i]);
while(!q.empty())
{
int t=q.front();
q.pop();
for(int i=0;i<26;i++)
if(c[t][i]) fail[c[t][i]]=c[fail[t]][i],q.push(c[t][i]);
else c[t][i]=c[fail[t]][i];
}
for(int i=1;i<=cnt;i++)
link(i,fail[i]);
dfs(0,0);
}
int lowbit(int x)
{
return x&(-x);
}
void change(int x,int y)
{
for(;x<=Index;x+=lowbit(x))
bit[x]+=y;
}
int ask(int x)
{
int sum=0;
for(;x>0;x-=lowbit(x)) sum+=bit[x];
return sum;
}
int main()
{
ins();
build();
m=read();
for(int i=1;i<=m;i++)
{
int x=read(),y=read();
g[y].push_back(node(x,i));
}
for(int i=0,now=0,j=0;i<len;i++)
{
if(s[i]=='P')
{
j++;
for(int k=0;k<g[j].size();k++)
{
int x=g[j][k].x,id=g[j][k].id;
ans[id]=ask(out[a[x]])-ask(dfn[a[x]]-1);
}
}
if(s[i]=='B')
{
change(dfn[now],-1);
now=fa[now];
}
if(s[i]>='a' && s[i]<='z')
{
now=c[now][s[i]-'a'];
change(dfn[now],1);
}
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
}