BZOJ-3881：Divljak (AC自动机+DFS序+树链求并+树状数组)

Alice有n个字符串S_1,S_2...S_n，Bob有一个字符串集合T，一开始集合是空的。

接下来会发生q个操作，操作有两种形式：

“1 P”，Bob往自己的集合里添加了一个字符串P。

“2 x”，Alice询问Bob，集合T中有多少个字符串包含串S_x。（我们称串A包含串B，当且仅当B是A的子串）

Bob遇到了困难，需要你的帮助。

 

 

Input

第1行，一个数n；

接下来n行，每行一个字符串表示S_i；

下一行，一个数q；

接下来q行，每行一个操作，格式见题目描述。

Output

对于每一个Alice的询问，帮Bob输出答案。

Sample Input

3
a
bc
abc
5
1 abca
2 1
1 bca
2 2
2 3

Sample Output

1
2
1

Hint

【数据范围】

1 <= n,q <= 100000；

Alice和Bob拥有的字符串长度之和各自都不会超过 2000000；

字符串都由小写英文字母组成。

之前就看过此题，但是不会树链求并，就搁着了。现在回来看还是可以做的。

Alice在字符串集合为S，Bob的字符串集合为T。先对Alice的集合建立AC自动机，得到Fail树。对于T每次新加的字符串Str，Str在自动机上跑。能跑到的所有 位置，在Fail树上对应的位置到根的所有点都要加1，重复的不重复加，就涉及到去重，即用树链求并。

把跑到是所有位置对应的Fail树位置按DFS序排序，每一个位置+1，相邻的LCA位置处-1。那么求T中有几个串含有某Si时，即是求Si在Fail树上的子树值之和。这里倍增求LCA会超时。

这个人的代码风格和我一模一样哎（但是他的AC自动机写丑了）！！！因为是权限题，自己写一遍怕错了没法知晓，就假设我写了，去找个差不多的题写写，233。

#include<bits/stdc++.h>
using namespace std;

#define N 2002010
int ch[N][],fail[N],ins[N],cnt;
char s[N];

int head[N],next[N],end[N];
inline void addedge(int a,int b){
    static int q=;end[q]=b,next[q]=head[a],head[a]=q++;
}

int in[N],out[N],dep[N],tclock;
void dfs(int x){
    in[x]=++tclock;
    for(int j=head[x];j;j=next[j])
        dep[end[j]]=dep[x]+,dfs(end[j]);
    out[x]=tclock;
}

namespace Lca_system{
    int Seq[N<<],a[],M,ins[N],cnt;
    inline void build_dfs(int x){
        ins[x]=++cnt;
        Seq[cnt]=x;
        for(int j=head[x];j;j=next[j]){
            build_dfs(end[j]);
            Seq[++cnt]=x;
        }
    }
    inline int Min(int x,int y){
        if(x==-||y==-)return x==-?y:x;
        return dep[x]<dep[y]?x:y;
    }
    inline void init(){
        build_dfs();
        for(M=;M<cnt+;M<<=);
        memset(a,-,sizeof a);
        for(int i=;i<=cnt;++i)a[M+i]=Seq[i];
        for(int i=M-;i>=;--i)a[i]=Min(a[*i],a[*i+]);
    }
    inline int ask(int tl,int tr){
        int re=-;
        for(tl+=M-,tr+=M+;tl^tr^;tl>>=,tr>>=){
            if(~tl&)re=Min(re,a[tl^]);
            if(tr&)re=Min(re,a[tr^]);
        }
        return re;
    }
    inline int lca(int x,int y){
        x=ins[x],y=ins[y];
        if(x>y)swap(x,y);
        return ask(x,y);
    }
}

int seq[N],num;

int A[N];
inline void modify(int x,int c){
    for(;x<=cnt+;x+=x&-x)A[x]+=c;
}
inline int ask(int x){
    int re=;for(;x;x-=x&-x)re+=A[x];return re;
}

inline bool cmp(const int&x,const int&y){return in[x]<in[y];}

inline int git(){
    int c,re;
    while(!isdigit(c=getchar()));
    re=c-'';
    while(isdigit(c=getchar()))re=(re<<)+(re<<)+c-'';
    return re;
}
char buf[*],*o=buf;
inline void print(int x){
    static int s[];int top=;
    if(!x)*o++=;else{for(;x;x/=)s[++top]=x%;for(int i=top;i>=;--i)*o++=+s[i];}
    *o++='\n';
}
int main(){
    int n=git();
    register int i,j;

    int len,p;
    for(i=;i<=n;++i){
        scanf("%s",s);
        len=strlen(s),p=;
        for(j=;j<len;++j){
            if(!ch[p][s[j]-'a'])ch[p][s[j]-'a']=++cnt;
            p=ch[p][s[j]-'a'];
        }
        ins[i]=p;
    }

    queue<int>q;
    for(i=;i<;++i) if(ch[][i]) q.push(ch[][i]);
    int u,v,r;
    while(!q.empty()){
        u=q.front(),q.pop();
        for(i=;i<;++i)if((v=ch[u][i])){
            q.push(v);
            for(r=fail[u];r&&!ch[r][i];r=fail[r]);
            fail[v]=ch[r][i];
        }
    }

    for(i=;i<=cnt;++i) addedge(fail[i],i);
    dep[]=,dfs();
    Lca_system::init();

    int Q=git();

    int qte,x;
    while(Q--){
        qte=git();
        if(qte==){
            scanf("%s",s);
            len=strlen(s),p=,num=;
            for(i=;i<len;++i){
                while(p&&!ch[p][s[i]-'a'])p=fail[p];
                p=ch[p][s[i]-'a'];
                if(p) seq[++num]=p;
            }
            sort(seq+,seq+num+,cmp);
            for(i=;i<=num;++i)modify(in[seq[i]],);
            for(i=;i<num;++i)modify(in[Lca_system::lca(seq[i],seq[i+])],-);
        }
        else{
            x=git();
            print(ask(out[ins[x]])-ask(in[ins[x]]-));
        }
    }

    fwrite(buf,,o-buf,stdout);
    return ;
}