CF163E e-Government

Posted 2021-02-16 asuldb

题目

两个(log)的树状数组套树剖？

我们对于给出的(n)个模式串建立(AC)自动机，之后对于每一个询问串直接丢上去匹配

如果这里是暴力的话，我们直接一路跳(fail)累加作为结束位置还没有被删除的点就好了

我们考虑一个快点的方式，树剖

把(fail)树建出来，直接在上面树剖维护就好了

由于只是单点修改，我们树状数组就好了

尽管是两个(log)，但毕竟树剖和树状数组都是出了名的小常数，还是能跑过(1e6)的

代码

#include<cstdio>
#include<queue>
#include<cstring>
#include<iostream>
#include<algorithm>
#define re register
#define lb(x) ((x)&(-x))
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=1e6+5;
struct E{int v,nxt;}e[maxn];
int n,m,cnt,__,num;
char S[maxn];
int c[maxn],dfn[maxn],sum[maxn],top[maxn],head[maxn];
int son[maxn][26],fa[maxn],pos[maxn],val[maxn],s[maxn];
inline void add_edge(int x,int y) {
    e[++num].v=y;e[num].nxt=head[x];head[x]=num;
}
inline void ins(int o) {
    scanf("%s",S+1);int len=strlen(S+1);
    int now=0;
    for(re int i=1;i<=len;i++) {
        if(!son[now][S[i]-'a']) son[now][S[i]-'a']=++cnt;
        now=son[now][S[i]-'a'];
    }
    pos[o]=now;
}
void dfs1(int x) {
    int maxx=-1;s[x]=-1,sum[x]=1;
    for(re int i=head[x];i;i=e[i].nxt) {
        dfs1(e[i].v);
        sum[x]+=sum[e[i].v];
        if(sum[e[i].v]>maxx) maxx=sum[e[i].v],s[x]=e[i].v;
    }
}
void dfs2(int x,int topf) {
    top[x]=topf,dfn[x]=++__;
    if(s[x]==-1) return;
    dfs2(s[x],topf);
    for(re int i=head[x];i;i=e[i].nxt)
    if(s[x]!=e[i].v) dfs2(e[i].v,e[i].v); 
}
inline void Build() {
    std::queue<int> q;
    for(re int i=0;i<26;i++) if(son[0][i]) q.push(son[0][i]);
    while(!q.empty()) {
        int k=q.front();q.pop();
        for(re int i=0;i<26;i++)
        if(son[k][i]) fa[son[k][i]]=son[fa[k]][i],q.push(son[k][i]);
            else son[k][i]=son[fa[k]][i];
    }
}
inline void add(int x,int v) {for(re int i=x;i<=cnt;i+=lb(i)) c[i]+=v;}
inline int ask(int x) {
    int now=0;
    for(re int i=x;i;i-=lb(i)) now+=c[i];
    return now;
}
inline int query(int l,int r) {return ask(r)-ask(l-1);}
inline int Query(int x) {
    int tot=0;
    while(top[x]) {
        tot+=query(dfn[top[x]],dfn[x]);
        x=fa[top[x]];
    }
    tot+=query(dfn[0],dfn[x]);
    return tot;
}
int main() {
    scanf("%d%d",&m,&n);
    for(re int i=1;i<=n;i++) ins(i);
    Build();
    for(re int i=1;i<=cnt;i++) add_edge(fa[i],i);
    dfs1(0);dfs2(0,0);cnt++;
    for(re int i=1;i<=n;i++) val[i]=1,add(dfn[pos[i]],1);
    while(m--) {
        scanf("%s",S);int len=strlen(S);
        if(S[0]=='+') {
            int x=0;
            for(re int i=1;i<len;i++) x=(x<<3)+(x<<1)+S[i]-48;
            if(val[x]) continue;
            add(dfn[pos[x]],1);val[x]=1;
        }
        if(S[0]=='-') {
            int x=0;
            for(re int i=1;i<len;i++) x=(x<<3)+(x<<1)+S[i]-48;
            if(!val[x]) continue;
            add(dfn[pos[x]],-1);val[x]=0;
        }
        if(S[0]=='?') {
            int now=0;int ans=0;
            for(re int i=1;i<len;i++) {
                now=son[now][S[i]-'a'];
                ans+=Query(now);
            }
            printf("%d
",ans);
        }
    }
    return 0;
}