bzoj3881 [Coci2015]Divljak（fail树+树链的并+BIT）

一开始给定模式串集合S，不断向文本串集合T添加字符串，多次询问T中有多少个串包含模式串Si。

我们首先把S集合的串用ACAM建出fail树，然后每次加入新串时我们就去ACAM上跑一遍，把到达的节点记下来。因为对每个串只贡献一次，所以这一次的贡献就是这若干个节点到根的若干条链的并。

怎么求链的并呢？

将所有节点按照DFS序排序
每个点到根的路径上的所有节点权值+1
相邻两个点的LCA到根的路径上的所有节点权值-1
即是树链的并

这些操作我们可以转化为单点修改和子树查询，用dfs序+BIT即可解决。

算内存真是算的我胆颤心惊qaq

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define N 2000010
#define M 100010
#define inf 0x3f3f3f3f
inline int read(){
    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;
}
int nn,n=0,son[N][26],fail[N],h[N],num=1,id[M],Log[N],rt;
int in[N],out[N],c[N],dfn=0,a[N],fa[N][21],dep[N];
char s[N];
struct edge{
    int to,next;
}data[N];
inline void adde(int x,int y){
    data[++num].to=y;data[num].next=h[x];h[x]=num;
}
inline void ins(char *s,int ii){
    int len=strlen(s+1),p=rt;
    for(int i=1;i<=len;++i){
        int &y=son[p][s[i]-'a'];if(!y) y=++n;p=y;
    }id[ii]=p;
}
inline void buildAC(){
    queue<int>q;q.push(rt);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=0;i<26;++i){
            int &y=son[x][i];
            if(!y){y=son[fail[x]][i];continue;}q.push(y);
            fail[y]=son[fail[x]][i];adde(fail[y],y);
        }
    }
}
inline void dfs(int x){
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }in[x]=++dfn;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;dep[y]=dep[x]+1;fa[y][0]=x;dfs(y);
    }out[x]=dfn;
}
inline void add(int x,int val){for(;x<=n;x+=x&-x) c[x]+=val;}
inline int ask(int x){int res=0;for(;x;x-=x&-x) res+=c[x];return res;}
inline bool cmp(int a,int b){return in[a]<in[b];}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline void gao(char *s){
    int len=strlen(s+1),p=rt,tot=0;
    for(int i=1;i<=len;++i){
        p=son[p][s[i]-'a'];a[++tot]=p;
    }sort(a+1,a+tot+1,cmp);add(in[a[1]],1);
    for(int i=2;i<=tot;++i)
        add(in[a[i]],1),add(in[lca(a[i],a[i-1])],-1);
}
int main(){
//  freopen("a.in","r",stdin);
    nn=read();rt=++n;for(int i=0;i<26;++i) son[0][i]=rt;Log[0]=-1;
    for(int i=1;i<=nn;++i) scanf("%s",s+1),ins(s,i);buildAC();
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;dfs(1);nn=read();
    while(nn--){
        int op=read();
        if(op==2){int x=id[read()];printf("%d\n",ask(out[x])-ask(in[x]-1));continue;}
        scanf("%s",s+1);gao(s);
    }return 0;
}