bzoj2434: [Noi2011]阿狸的打字机 trie+线段树

我们可以先按题目描述建出一个trie树，然后得到fall树，我们可以发现，fall树的子树里有#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
#define maxn 110000
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define getmid int mid=(l+r)>>1
int sum[maxn*8];
void pushup(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void add(int l,int r,int rt,int pos,int v)
{
if(l==r)
{
sum[rt]+=v;
return;
}
getmid;
if(pos<=mid) add(lson,pos,v);
else add(rson,pos,v);
pushup(rt);
}
int qsum(int l,int r,int rt,int x,int y)
{
if(x<=l&&r<=y)
{
return sum[rt];
}
getmid;
int res=0;
if(x<=mid) res+=qsum(lson,x,y);
if(y>mid) res+=qsum(rson,x,y);
return res;
}
int a[maxn][26],n,m,cnt=1,fa[maxn],point[maxn],pos[maxn];
char st[maxn];
int read()
{
char c;int res;
while(c=getchar(),c<'0'||c>'9');
res=c-'0';
while(c=getchar(),c>='0'&&c<='9')
res=res*10+c-'0';
return res;
}
void build()
{
for(int i=0;i<26;i++)
{
a[0][i]=1;
}
n=strlen(st+1);
int now=1,id=0;
for(int i=1;i<=n;i++)
{
if(st[i]=='P')
{
id++;
pos[id]=now;
}
else if(st[i]=='B') now=fa[now];
else
{
if(!a[now][st[i]-'a'])
{
cnt++;
a[now][st[i]-'a']=cnt;
fa[cnt]=now;
}
now=a[now][st[i]-'a'];
}
}
}
int q[maxn];
void acm()
{
int head=0,tail=0;
tail++;q[1]=1;
point[1]=0;
while(head<tail)
{
head++;
int now=q[head];
for(int i=0;i<26;i++)
{
if(!a[now][i]) continue;
int k=point[now];
while(!a[k][i]) k=point[k];
point[a[now][i]]=a[k][i];
tail++; q[tail]=a[now][i];
}
}
}
int en,first[maxn],next[maxn],to[maxn],ind;
void adde(int a,int b)
{
en++;
to[en]=b;next[en]=first[a];
first[a]=en;
}
int in[maxn],out[maxn];
void dfs(int x)
{
ind++;
in[x]=ind;
for(int i=first[x];i;i=next[i])
{
dfs(to[i]);
}
ind++;
out[x]=ind;
}
struct node
{
int to,next,v;
}que[100005];
int last[maxn],ans[maxn];
void solve()
{
add(1,cnt*2,1,in[0],1);
int now=1,id=0;
for(int i=1;i<=n;i++)
{
if(st[i]=='P')
{
id++;
for(int x=last[id];x;x=que[x].next)
{
int v=pos[que[x].to];
ans[x]=qsum(1,cnt*2,1,in[v],out[v]);
}
}
else if(st[i]=='B')
{
add(1,cnt*2,1,in[now],-1);
now=fa[now];
}
else
{
now=a[now][st[i]-'a'];
add(1,cnt*2,1,in[now],1);
}
}
}
int main()
{
scanf("%s",st+1);
build();
acm();
for(int i=1;i<=cnt;i++)
{
adde(point[i],i);
}
dfs(0);
m=read();
for(int i=1;i<=m;i++)
{
int x=read(),y=read();
que[i].next=last[y];
last[y]=i;
que[i].to=x;
}
solve();
for(int i=1;i<=m;i++)
{
printf("%d\n",ans[i]);
}
return 0;
}

几个节点，那么这个串就出现了多少次，由于是棵树，我们可以用的dfs序来维护。考虑在线维护的话，返回到历史k十分麻烦，只能用离线处理。