【bzoj3881】【Coci2015】Divljak

• 题解

  • 对$S$集合ac建自动机，把$T_{i}$放在里面跑，记录路径上的所有节点并对它们在fail树上求到root的树链并；
  • 这样就得到了$T_{i}$所有的子串；
  • 动态将$T_{i}$加入直接用树状数组维护子树和即可；
  • 另见：喵星球上的点名的ac自动机解法

  •  #include<bits/stdc++.h>
     #define rg register
     #define il inline
     using namespace std;
     const int N=;
     int n,m,cnt,ch[N][],fl[N],fa[N],head,tail,que[N],o=,hd[N],sz[N];
     int tp[N],dep[N],idx,st[N],ed[N],c[N],vis[N],id[N],tmp[N],tot,tt;
     char s[N];
     struct Edge{int v,nt;}E[N];
     il void adde(int u,int v){E[o]=(Edge){v,hd[u]};hd[u]=o++;}
     il char gc(){
         static char*p1,*p2,S[];
         if(p1==p2)p2=(p1=S)+fread(S,,,stdin);
         return(p1==p2)?EOF:*p1++;
     }
     il int rd(){
         int x=; char C=gc();
         while(C<''||C>'')C=gc();
         while(C>=''&&C<='')x=(x<<)+(x<<)+C-'',C=gc();
         return x;
     }
     il int gt(){
         char *p=s,C=gc();
         while(!isalpha(C))C=gc();
         while(isalpha(C))*p++=C,C=gc();
         return p - s;
     }
     il bool cmp(const int&a,const int&b){return st[a]<st[b];}
     il void ins(int now,int len){
         int u=,x;
         for(int i=;i<len;i++){
             if(!ch[u][x=s[i]-'a'])ch[u][x]=++cnt;
             u=ch[u][x];
         }
         sz[id[now]=u]=vis[u]=;
     }
     void get_fl(){
         head=tail=;
         for(rg int i=;i<;i++)if(ch[][i]){
             que[++tail]=ch[][i];
             if(vis[ch[][i]])adde(,ch[][i]);
         }
         while(head<tail){
             int u=que[++head];
             for(rg int i=;i<;i++){
                 int&v=ch[u][i];
                 if(!v){v=ch[fl[u]][i];continue;}
                 fl[v]=ch[fl[u]][i];
                 if(vis[fl[v]])fa[v]=fl[v];
                 else fa[v]=fa[fl[v]];
                 if(vis[v])adde(fa[v],v);
                 que[++tail]=v;
             }
         }
         for(int i=tail;i;i--)if(vis[que[i]])sz[fa[que[i]]]+=sz[que[i]];
     }
     il void dfs(int u,int T){
         st[u]=++idx;tp[u]=T;
         dep[u]=dep[fa[u]]+;
         int mx=-,son=;
         for(rg int i=hd[u];i;i=E[i].nt){if(sz[E[i].v]>mx)mx=sz[son=E[i].v];}
         if(son)dfs(son,T);
         for(rg int i=hd[u];i;i=E[i].nt){if(E[i].v!=son)dfs(E[i].v,E[i].v);}
         ed[u]=idx;
     }
     il void add(int x,int y){if(!x)return;for(;x<=idx;x+=x&-x)c[x]+=y;}
     il int ask(int x){int re=;for(;x;x-=x&-x)re+=c[x];return re;}
     il int lca(int u,int v){
         int tu=tp[u],tv=tp[v];
         while(tu!=tv){
             if(dep[tu]<dep[tv])v=fa[tv],tv=tp[v];
             else u=fa[tu],tu=tp[u];
         }
         return dep[u]<dep[v]?u:v;
     }
     void update(int len){
         tt=tot=;
         for(rg int i=,u=;i<len;i++){
             u=ch[u][s[i]-'a'];
             if(vis[u])tmp[++tt]=u;
             else if(fa[u])tmp[++tt]=fa[u];
         }
         if(!tt)return;
         sort(tmp+,tmp+tt+,cmp);
         tot=;
         for(rg int i=;i<=tt;i++){
             tot+=st[tmp[i]]>ed[tmp[tot]];
             tmp[tot]=tmp[i];
         }
         add(st[tmp[]],);
         for(rg int i=;i<=tot;i++){
             add(st[lca(tmp[i],tmp[i-])],-);
             add(st[tmp[i]],);
         }
     }
     int main(){
         #ifndef ONLINE_JUDGE
         freopen("bzoj3881.in","r",stdin);
         freopen("bzoj3881.out","w",stdout);
         #endif
         n=rd();
         for(rg int i=;i<=n;i++){ins(i,gt());}
         get_fl();
         dfs(,);
         m=rd();
         for(rg int i=,x,y;i<=m;i++){
             x=rd();
             if(x==)update(gt());
             else y=rd(),printf("%d\n",ask(ed[id[y]])-ask(st[id[y]]-));
         }
         return ;
     }

    bzoj3881