bzoj 2434: [Noi2011]阿狸的打字机

 #include<cstdio>
#include<iostream>
#include<cstring>
#define M 100008
using namespace std;
char ch[M];
int cnt=,pos[M],now,fa[M],fail[M],q[M],head[M],next[M],u[M],cnt1,m,n;
int a[M][],head1[M],next1[M],u1[M],T,l[M],r[M],an[M],su[*M];
void build()
{
int h=,t=;
q[]=;
for(;h<t;)
{
int p=q[++h];
for(int i=;i<=;i++)
if(a[p][i])
{
int now=fail[p];
q[++t]=a[p][i];
for(;!a[now][i];now=fail[now]);
fail[a[p][i]]=a[now][i];
}
}
return;
}
void jia(int a1,int a2)
{
cnt1++;
u[cnt1]=a2;
next[cnt1]=head[a1];
head[a1]=cnt1;
return;
}
void dfs(int a1)
{
l[a1]=++T;
for(int i=head[a1];i;i=next[i])
dfs(u[i]);
r[a1]=++T;
}
int xun(int a1)
{
int sum=;
for(int i=a1;i;i-=i&-i)
sum+=su[i];
return sum;
}
void add(int a1,int a2)
{
for(int i=a1;i<=T;i+=i&-i)
su[i]+=a2;
}
int main()
{
scanf("%s",ch+);
now=;
n=strlen(ch+);
for(int i=;i<=n;i++)
if(ch[i]=='P')
{
pos[]++;
pos[pos[]]=now;
}
else if(ch[i]=='B')
now=fa[now];
else
{
if(!a[now][ch[i]-])
{
cnt++;
a[now][ch[i]-]=cnt;
fa[cnt]=now;
}
now=a[now][ch[i]-];
}
for(int i=;i<=;i++)
a[][i]=;
build();
for(int i=;i<=cnt;i++)
jia(fail[i],i);
scanf("%d",&m);
for(int i=;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
u1[i]=x;
next1[i]=head1[y];
head1[y]=i;
}
dfs();
now=;
pos[]=;
for(int i=;i<=n;i++)
if(ch[i]=='P')
{
pos[]++;
for(int i=head1[pos[]];i;i=next1[i])
an[i]=xun(r[pos[u1[i]]])-xun(l[pos[u1[i]]]-);
}
else if(ch[i]=='B')
{
add(l[now],-);
now=fa[now];
}
else
{
now=a[now][ch[i]-];
add(l[now],);
}
for(int i=;i<=m;i++)
printf("%d\n",an[i]);
return ;
}

AC自动机 根据dfs序建立fail树(就是将失败指针反向),每次查询x子树中有多少y的节点。

上一篇:Reprint: ADB is Not Recognized as an internal or external command Fix


下一篇:centos6.5虚拟机安装后,没有iptables配置文件