有些算法学了很久没有复习,今天重刷了两道后缀数组的题,然后想找到AC自动机的题目,毕竟以前也没写过几遍。
然后尴尬的发现去年noi的阿狸的打字机还没写,去年暑假因为不懂AC自动机,没有去动它,后来就一直忘记了=。=!
不说废话了。
将所有输出的单词存入AC自动机,由于题目特殊,建起来比一般的AC自动机要简单。
考虑打裸,对于询问(i,j),枚举j单词上的每一个点,如果该点可以通过fail指针指向i的末尾,那么ans++;
这样当然是不能AC的,注意到,如果把fail指针反向,然后建立一颗fail树,那么,从j上的点可以通过fail指针走到i的数量,等价于在fail tree上,以i根的子树中,
有多少个j单词上的点。
这样,就不难想到,如果我们假设fail tree 上,之后j单词上的点有权值,那么我们可以轻易的用dfs序列+树状数组/线段树统计出来。那么,可以考虑dfs这个trie, dfs过程中修改fail tree 上的权值,很容易可以做到dfs 到j单词末尾是,仅有j单词上的点有权值 ,这样用离线的方法可以处理所有询问。
AC自动机貌似不会怎么出错,一遍过了。
# include <cstdlib>
# include <cstdio>
# include <cmath>
# include <cstring>
using namespace std;
const int maxn = 100000+5;
struct list
{
int top, linke[maxn*2+10], sum[maxn*2+10], next[maxn*2+10];
void link(int x, int y)
{
++top; next[top]= linke[x]; linke[x]=top; sum[top]=y;
}
}ans, edge;
int n, len;
int tot,end[maxn], fail[maxn], map[maxn][26], pre[maxn];
char s[maxn];
int l,r,que[maxn],st[maxn],have,ask[maxn], sol[maxn];
int head[maxn], tail[maxn], go[maxn], c[maxn];
void read()
{
int i,x,y;
scanf("%s\n", s+1);
scanf("%d", &n);
for (i = 1; i <= n; i++)
{
scanf("%d%d", &x, &y);
ans.link(y, x);
}
}
void build_AC()
{
int i,now=1; tot=1;len= strlen(s+1);
for (i = 1; i <= len; i++)
{
if (s[i]=='P') end[now]=++have, ask[have]=now;
else if (s[i]=='B') now=pre[now];
else now= map[now][s[i]-'a']==0? pre[++tot]=now,map[now][s[i]-'a']=tot: map[now][s[i]-'a'];
}
for (i = 0; i <= 25; i++)
if (map[1][i]) que[++r]=map[1][i], fail[que[r]]=1, edge.link(1, que[r]);
for (l=1;l<= r;l++)
for (i=0; i<=25; i++)
if (map[que[l]][i])
{
que[++r]=map[que[l]][i];
for (now=fail[que[l]]; now!=1&& map[now][i]==0;)now=fail[now];
fail[que[r]]= map[now][i]?map[now][i]:now;
edge.link(fail[que[r]], que[r]);
}
}
void pdfs(int now)
{
st[++st[0]]=now; head[now]=st[0]; go[now]=st[0];
for (int ke= edge.linke[now]; ke!=0; ke=edge.next[ke])
pdfs(edge.sum[ke]);
tail[now]=st[0];
}
void modify(int x, int d)
{
for (;x<=st[0];x+= x&(-x))
c[x]+= d;
}
int query(int x)
{
int ask=0;
for (;x>0;x-= x&(-x))
ask+= c[x];
return ask;
}
void work()
{
int i,now=1,ks;
for (i=1; i <= len; i++)
{
if (s[i]=='P');
else if (s[i]=='B') modify(go[now], -1),now=pre[now];
else now= map[now][s[i]-'a'], modify(go[now], 1);
if (ks=end[now])
{
for (int ke=ans.linke[ks]; ke!= 0; ke=ans.next[ke])
sol[ke]= query(tail[ask[ans.sum[ke]]])- query(head[ask[ans.sum[ke]]]-1);
end[now]=0;
};
}
}
int main()
{
//freopen("type.in", "r", stdin);
//freopen("type.out", "w", stdout);
read();
build_AC();
pdfs(1);
work();
for (int i = 1; i <= n; i++) printf("%d\n", sol[i]);
return 0;
}