description
维护一个字符串集合,支持加入一个串,删除一个串,查询集合中的串在给出串中出现了多少次。
强制在线。
sol
二进制分组。维护(log)个(mbox{AC})自动机,每插入一个串后依次合并,并重建(mbox{fail}),保证每个大小的(mbox{AC})自动机只存在一个。
有删除操作?注意到这里的答案是具有可减性的,所以再开一个二进制分组表示已删除的串,查询时就用两个答案相减即可。
code
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int gi(){
int x=0,w=1;char ch=getchar();
while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
if (ch=='-') w=0,ch=getchar();
while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
return w?x:-x;
}
const int N = 3e5+5;
struct AC{
int son[26][N],end[N],tr[26][N],fa[N],cnt[N],tot;
int rt[N],sz[N],top;queue<int>Q;
void getfail(int RT){
for (int i=0;i<26;++i)
if (son[i][RT]) fa[tr[i][RT]=son[i][RT]]=RT,Q.push(tr[i][RT]);
else tr[i][RT]=RT;
while (!Q.empty()){
int u=Q.front();Q.pop();
for (int i=0;i<26;++i)
if (son[i][u]){
tr[i][u]=son[i][u],fa[tr[i][u]]=tr[i][fa[u]];
Q.push(tr[i][u]);
}else tr[i][u]=tr[i][fa[u]];
cnt[u]=end[u]+cnt[fa[u]];
}
}
int merge(int x,int y){
if (!x||!y) return x|y;
end[x]+=end[y];
for (int i=0;i<26;++i) son[i][x]=merge(son[i][x],son[i][y]);
return x;
}
void insert(char *s,int n){
rt[++top]=++tot;sz[top]=1;int x=rt[top];
for (int i=1;i<=n;++i){
if (!son[s[i]-'a'][x]) son[s[i]-'a'][x]=++tot;
x=son[s[i]-'a'][x];
}
end[x]=1;
while (sz[top]==sz[top-1]){
--top;
rt[top]=merge(rt[top],rt[top+1]);sz[top]+=sz[top+1];
}
getfail(rt[top]);
}
int query(char *s,int n){
int res=0;
for (int i=1;i<=top;++i)
for (int j=1,x=rt[i];j<=n;++j)
x=tr[s[j]-'a'][x],res+=cnt[x];
return res;
}
}T1,T2;
char s[N];
int main(){
int m=gi();while (m--){
int op=gi();scanf("%s",s+1);int n=strlen(s+1);
if (op==1) T1.insert(s,n);
if (op==2) T2.insert(s,n);
if (op==3) printf("%d
",T1.query(s,n)-T2.query(s,n)),fflush(stdout);
}
return 0;
}