【bzoj3881】【Coci2015】Divljak

  •  题解

    • 对$S$集合ac建自动机,把$T_{i}$放在里面跑,记录路径上的所有节点并对它们在fail树上求到root的树链并;
    • 这样就得到了$T_{i}$所有的子串;
    • 动态将$T_{i}$加入直接用树状数组维护子树和即可;
    • 另见:喵星球上的点名的ac自动机解法
    •   1 #include<bits/stdc++.h>
        2 #define rg register
        3 #define il inline 
        4 using namespace std;
        5 const int N=2000010;
        6 int n,m,cnt,ch[N][26],fl[N],fa[N],head,tail,que[N],o=1,hd[N],sz[N];
        7 int tp[N],dep[N],idx,st[N],ed[N],c[N],vis[N],id[N],tmp[N],tot,tt;
        8 char s[N];
        9 struct Edge{int v,nt;}E[N];
       10 il void adde(int u,int v){E[o]=(Edge){v,hd[u]};hd[u]=o++;}
       11 il char gc(){
       12     static char*p1,*p2,S[1000000];
       13     if(p1==p2)p2=(p1=S)+fread(S,1,1000000,stdin);
       14     return(p1==p2)?EOF:*p1++;
       15 }
       16 il int rd(){
       17     int x=0; char C=gc();
       18     while(C<'0'||C>'9')C=gc();
       19     while(C>='0'&&C<='9')x=(x<<1)+(x<<3)+C-'0',C=gc();
       20     return x;
       21 }
       22 il int gt(){
       23     char *p=s,C=gc();
       24     while(!isalpha(C))C=gc();
       25     while(isalpha(C))*p++=C,C=gc();
       26     return p - s;
       27 }
       28 il bool cmp(const int&a,const int&b){return st[a]<st[b];}
       29 il void ins(int now,int len){
       30     int u=0,x;
       31     for(int i=0;i<len;i++){
       32         if(!ch[u][x=s[i]-'a'])ch[u][x]=++cnt;
       33         u=ch[u][x];
       34     }
       35     sz[id[now]=u]=vis[u]=1;
       36 }
       37 void get_fl(){
       38     head=tail=0;
       39     for(rg int i=0;i<26;i++)if(ch[0][i]){
       40         que[++tail]=ch[0][i];
       41         if(vis[ch[0][i]])adde(0,ch[0][i]);
       42     }
       43     while(head<tail){
       44         int u=que[++head];
       45         for(rg int i=0;i<26;i++){
       46             int&v=ch[u][i];
       47             if(!v){v=ch[fl[u]][i];continue;}
       48             fl[v]=ch[fl[u]][i];
       49             if(vis[fl[v]])fa[v]=fl[v];
       50             else fa[v]=fa[fl[v]];
       51             if(vis[v])adde(fa[v],v);
       52             que[++tail]=v;
       53         }
       54     }
       55     for(int i=tail;i;i--)if(vis[que[i]])sz[fa[que[i]]]+=sz[que[i]];
       56 }
       57 il void dfs(int u,int T){
       58     st[u]=++idx;tp[u]=T;
       59     dep[u]=dep[fa[u]]+1;
       60     int mx=-1,son=0;
       61     for(rg int i=hd[u];i;i=E[i].nt){if(sz[E[i].v]>mx)mx=sz[son=E[i].v];}
       62     if(son)dfs(son,T); 
       63     for(rg int i=hd[u];i;i=E[i].nt){if(E[i].v!=son)dfs(E[i].v,E[i].v);}
       64     ed[u]=idx;
       65 }
       66 il void add(int x,int y){if(!x)return;for(;x<=idx;x+=x&-x)c[x]+=y;}
       67 il int ask(int x){int re=0;for(;x;x-=x&-x)re+=c[x];return re;}
       68 il int lca(int u,int v){
       69     int tu=tp[u],tv=tp[v];
       70     while(tu!=tv){
       71         if(dep[tu]<dep[tv])v=fa[tv],tv=tp[v];
       72         else u=fa[tu],tu=tp[u];
       73     }
       74     return dep[u]<dep[v]?u:v;
       75 }
       76 void update(int len){
       77     tt=tot=0;
       78     for(rg int i=0,u=0;i<len;i++){
       79         u=ch[u][s[i]-'a'];
       80         if(vis[u])tmp[++tt]=u;
       81         else if(fa[u])tmp[++tt]=fa[u];
       82     }
       83     if(!tt)return;
       84     sort(tmp+1,tmp+tt+1,cmp);
       85     tot=1;
       86     for(rg int i=2;i<=tt;i++){
       87         tot+=st[tmp[i]]>ed[tmp[tot]];
       88         tmp[tot]=tmp[i];
       89     }
       90     add(st[tmp[1]],1);
       91     for(rg int i=2;i<=tot;i++){
       92         add(st[lca(tmp[i],tmp[i-1])],-1);
       93         add(st[tmp[i]],1);
       94     }
       95 }
       96 int main(){
       97     #ifndef ONLINE_JUDGE
       98     freopen("bzoj3881.in","r",stdin);
       99     freopen("bzoj3881.out","w",stdout);
      100     #endif 
      101     n=rd();
      102     for(rg int i=1;i<=n;i++){ins(i,gt());}
      103     get_fl();
      104     dfs(0,0);
      105     m=rd();
      106     for(rg int i=1,x,y;i<=m;i++){
      107         x=rd();
      108         if(x==1)update(gt());
      109         else y=rd(),printf("%d\n",ask(ed[id[y]])-ask(st[id[y]]-1));
      110     }
      111     return 0;
      112 }
      bzoj3881

       

posted @ 2019-01-09 07:13  大米饼  阅读(369)  评论(0编辑  收藏  举报