[luogu2414 NOI2011]阿狸的打字机 (AC自动机)

[luogu2414 NOI2011]阿狸的打字机 (AC自动机)

传送门

Solution

我们知道AC自动机上如果有一点A的fail[A]->B那么B为A的一个后缀

那么我们的问题\((x,y)\)就变为在y中有多少个点直接或间接连向x的终止节点

如果写暴力的话就是遍历y的所有点，对于每个点，我们往上暴跳fail，如果遇到x单词的结尾就ans+1

然而我可以利用dfs序和数据结构优美的完成

首先我们求出fail树（树边与fail相反）的dfs序，并将询问按y排序，然后从头扫读入的字符串，扫的时候给每个点对应的dfs序位置+1。
每当扫完一个字符串时，求出这个字符串作为y的贡献。
具体来说就是找到这个字符串对应的x，对每一个x的终止节点，求出子树的权值和即可

Code

//By Menteur_Hxy
#include <queue>
#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define fi first
#define se second
#define Re register 
#define Ms(a,b) memset(a,(b),sizeof(a))
#define Ee(i,u) for(Re int i=head[u];i;i=nxt[i])
#define Fo(i,a,b) for(Re int i=(a),_=(b);i<=_;i++)
#define Ro(i,a,b) for(Re int i=(b),_=(a);i>=_;i--)
#define Add(a,b) nxt[++cnt]=head[a],to[cnt]=b,head[a]=cnt
using namespace std;
typedef pair<int,int> PII;

inline int read() {
    int x=0,f=1;char c=getchar();
    while(!isdigit(c)) {if(c=='-')f=-f;c=getchar();}
    while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();
    return x*f;
}

const int L=1e5+10;
int m,tot,cnt;
int str[L],nxt[L],to[L],head[L],dfn[L],low[L],ans[L];
char ch[L];
vector <PII> As[L];

queue <int> Q;
struct AC_Automation{
    int nd[L][26],fail[L],fa[L],ret;
    AC_Automation() {Ms(nd,0);Ms(fail,0);Ms(fa,0);ret=0;}
    void ins(char *s) {
        int len=strlen(s),now=0,*p;
        Fo(i,0,len-1) {
            if(s[i]=='P') str[++tot]=now;
            else {
                if(s[i]=='B') now=fa[now];
                else {p=&nd[now][s[i]-'a'];if(!*p)*p=++ret;fa[*p]=now;now=*p;}
            }
        } 
    }
    void build() {
        Fo(i,0,25) if(nd[0][i]) fail[nd[0][i]]=0,Q.push(nd[0][i]);
        while(!Q.empty()) {
            int now=Q.front(),*p;Q.pop();
            Fo(i,0,25) if(*(p=&nd[now][i])) fail[*p]=nd[fail[now]][i],Q.push(*p);
                else *p=nd[fail[now]][i];
        }
    }
}AC;

struct BIT{
    int da[L];
    BIT() {clear();}
    void clear() {Ms(da,0);}
    void add(int x,int k) {for(;x<=L;x+=x&-x)da[x]+=k;}
    int qry(int x) {int res=0;for(;x;x-=x&-x)res+=da[x];return res;}
}T;

void dfs(int u) {
    dfn[u]=++tot;
    Ee(i,u) dfs(to[i]);
    low[u]=tot;
}

int main() {
    scanf("%s",ch); AC.ins(ch); AC.build();
    m=read();
    Fo(i,1,m) {
        int x=read(),y=read();
        As[y].push_back(PII(x,i));
    }
    Fo(i,1,AC.ret) Add(AC.fail[i],i); dfs(0);
    T.add(dfn[0],1); int now=0,tmp=0;
    for(Re int i=0;ch[i];i++) {
        if(ch[i]=='P') {
            tmp++;
            Fo(j,0,As[tmp].size()-1) {
                int x=str[As[tmp][j].fi];
                ans[As[tmp][j].se]=T.qry(low[x])-T.qry(dfn[x]-1);
            }
        } else if(ch[i]=='B') T.add(dfn[now],-1),now=AC.fa[now];
        else now=AC.nd[now][ch[i]-'a'],T.add(dfn[now],1);
    }
    Fo(i,1,m) printf("%d\n",ans[i]);
    return 0;
}

posted @ 2018-10-08 12:01 Menteur_Hxy 阅读( ...) 评论( ...) 编辑 收藏