2434: [Noi2011]阿狸的打字机|Ac自动机

最新推荐文章于 2022-07-19 11:51:23 发布

ws_yzy

最新推荐文章于 2022-07-19 11:51:23 发布

阅读量396

点赞数 1

分类专栏： AC自动机文章标签： Ac自动机

本文链接：https://blog.csdn.net/ws_yzy/article/details/50887364

版权

AC自动机专栏收录该内容

1 篇文章 0 订阅

订阅专栏

先建出AC自动机，求出fail数组，第x字符串在第y个字符串中出现的次数可以这样统计：从y到root的fail指针能找到x的次数，然后建出fail树之后也就是y到root在x的子树中的个数。找出dfs序再扫一遍字符串用树状数组维护一下。
代码长度2500B真是感人。。

#include<set>
#include<map>
#include<ctime>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define  N 200005
#define mx 1e9
using namespace std;
int sc()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
char s[N];
int Head[N],Nxt[N],Lst[N];
int head[N],nxt[N],lst[N];
int ch[N][26],fail[N],fa[N],pos[N];
int S[N],T[N],q[N],tr[N],ans[N];
int n,cnt=1,m,Q,tot,E,Tot;
void build()
{
    int now=1;
    for(int i=0;i<n;i++)
        if(s[i]=='P')
            pos[++m]=now;
        else if(s[i]=='B') now=fa[now];
        else 
        {
            if(!ch[now][s[i]-'a'])ch[now][s[i]-'a']=++cnt;
            fa[ch[now][s[i]-'a']]=now;
            now=ch[now][s[i]-'a'];
        }
}
void build_fail()
{
    fail[q[1]=1]=0;
    for(int l=1,r=2;l<r;l++)
    {
        int x=q[l];
        for(int i=0;i<26;i++)
            if(ch[x][i])
            {
                int k=fail[x];
                while(!ch[k][i])k=fail[k];
                fail[ch[x][i]]=ch[k][i];
                q[r++]=ch[x][i];
            }
    }
}
void change(int x,int f)
{
    for(;x<=E;x+=x&-x)tr[x]+=f;
}
int ask(int x)
{
    int ans=0;
    for(;x;x-=x&-x)ans+=tr[x];
    return ans;
}
void insert(int x,int y)
{
    lst[++tot]=y;nxt[tot]=head[x];head[x]=tot;
}
void Insert(int x,int y)
{
    Lst[++Tot]=y;Nxt[Tot]=Head[x];Head[x]=Tot;
}
void dfs(int x)
{
    S[x]=++E;
    for(int i=head[x];i;i=nxt[i])
        dfs(lst[i]);
    T[x]=E;
}
void solve()
{
    int now=1,x=0;
    for(int i=0;i<n;i++)
        if(s[i]=='P')
        {
            x++;
            for(int p=Head[x];p;p=Nxt[p])
            {
                int t=pos[Lst[p]];
                ans[p]=ask(T[t])-ask(S[t]-1);
            }
        }
        else if(s[i]=='B')change(S[now],-1),now=fa[now];
        else now=ch[now][s[i]-'a'],change(S[now],1);
}
int main()
{
    for(int i=0;i<26;i++)ch[0][i]=1;
    scanf("%s",s);
    n=strlen(s);
    build();
    build_fail();
    for(int i=1;i<=cnt;i++)
        insert(fail[i],i);
    Q=sc();
    for(int i=1;i<=Q;i++)
    {
        int x=sc(),y=sc();
        Insert(y,x);
    }
    dfs(0);
    solve();
    for(int i=1;i<=Q;i++)
        printf("%d\n",ans[i]);
    return 0;
}