对Alice的所有串构造AC自动机,并建出Fail树
每当Bob添加一个串时,在AC自动机上走,每走到一个点,就把它到根路径上所有点的答案+1
需要注意的是每次操作,相同的点只能被加一次
所以在需要操作的点构成的虚树上进行修改操作,避免重复修改
对于修改,x到根路径上所有点答案+1等价于在x点打上一个标记,对于所有祖先都有效
由此将问题转化为单点修改,子树查询
树状数组维护
注意到关键点其实只有n个,所以可以一开始先对这n个点求一次虚树来降低常数
时间复杂度$O(L\log L)$,L为所有串的长度之和
#include<cstdio> #include<algorithm> #define N 2000010 using namespace std; int tot,son[N][26],f[N],v[N],nxt[N],g[N],ed,q[N],h=1,t=1; int size[N],heavy[N],d[N],top[N],st[N],en[N],dfn,bit[N],fin[100010],fir[N],fa[N]; int m,a[N],b[N];bool vis[N],have[N]; int n,Q,i,k,j,x,op,l; int need[N],cn; char s[N],c; inline void read(int&a){while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';} inline void readstr(){for(l=0;((c=getchar())>='a')&&(c<='z');s[l++]=c);} inline bool cmp(int x,int y){return st[x]<st[y];} inline void ins(int p){ for(int x=0,i=0,w;i<l;i++){ if(!son[x][w=s[i]-'a'])son[x][w]=++tot;x=son[x][w]; if(i==l-1)fin[p]=x; } } inline void add(int x,int y){fa[y]=x;v[++ed]=y;nxt[ed]=g[x];g[x]=ed;} void dfs(int x){ st[x]=++dfn;d[x]=d[fa[x]]+1;size[x]=1;heavy[x]=-1; for(int i=g[x];i;i=nxt[i])if(v[i]!=fa[x]){ dfs(v[i]);size[x]+=size[v[i]]; if(heavy[x]<0||size[v[i]]>size[heavy[x]])heavy[x]=v[i]; } en[x]=dfn; } void dfs2(int x,int y){ top[x]=y; if(~heavy[x])dfs2(heavy[x],y); for(int i=g[x];i;i=nxt[i])if(v[i]!=heavy[x]&&v[i]!=fa[x])dfs2(v[i],v[i]); } void dfs3(int x,int y){ if(vis[x])y=x; fir[x]=y; for(int i=g[x];i;i=nxt[i])if(v[i]!=fa[x])dfs3(v[i],y); } inline int lca(int x,int y){ for(;top[x]!=top[y];x=fa[top[x]])if(d[top[x]]<d[top[y]])swap(x,y); return d[x]<d[y]?x:y; } inline void modify(int x,int y){for(;x<=dfn;x+=x&-x)bit[x]+=y;} inline int sum(int x){t=0;for(;x;x-=x&-x)t+=bit[x];return t;} inline void match(){ for(int x=0,i=m=0,w;i<l;i++){ w=s[i]-'a'; while(x&&!son[x][w])x=f[x];x=son[x][w]; if(!vis[fir[x]])vis[a[++m]=fir[x]]=1; } } int main(){ for(read(n),i=1;i<=n;i++)readstr(),ins(i); f[0]=-1; while(h<=t)for(x=q[h++],i=0;i<26;i++)if(son[x][i])for(q[++t]=son[x][i],j=f[x];~j;j=f[j])if(x&&son[j][i]){f[son[x][i]]=son[j][i];break;} for(i=1;i<=tot;i++)add(f[i],i); dfs(0),dfs2(0,0); vis[0]=m=1; for(i=1;i<=n;i++)if(!vis[fin[i]])vis[a[++m]=fin[i]]=1; sort(a+1,a+m+1,cmp); for(cn=m,i=1;i<m;i++)if(!vis[x=lca(a[i],a[i+1])])vis[a[++cn]=x]=1; dfs3(0,0); for(ed=i=0;i<=tot;i++)g[i]=0; m=cn,sort(a+1,a+m+1,cmp); for(q[t=1]=0,i=2;i<=m;q[++t]=a[i++]){ while(st[a[i]]<st[q[t]]||en[a[i]]>en[q[t]])t--; add(q[t],a[i]); } for(i=1;i<=m;i++)vis[a[i]]=0; dfs(dfn=cn=0);dfs2(0,0); read(Q); while(Q--){ read(op); if(op==1){ readstr(); match(); sort(a+1,a+m+1,cmp); for(tot=m,i=1;i<m;i++)if(!vis[x=lca(a[i],a[i+1])])vis[a[++tot]=x]=1; m=tot,sort(a+1,a+m+1,cmp); if(!have[x=a[1]])have[x]=1,b[x]=0,need[++cn]=x;b[x]++; for(q[t=1]=a[1],i=2;i<=m;q[++t]=a[i++]){ while(st[a[i]]<st[q[t]]||en[a[i]]>en[q[t]])t--; if(!have[x=q[t]])have[x]=1,b[x]=0,need[++cn]=x;b[x]--; if(!have[x=a[i]])have[x]=1,b[x]=0,need[++cn]=x;b[x]++; } for(i=1;i<=m;i++)vis[a[i]]=0; }else{ if(cn){ for(i=1;i<=cn;i++)x=need[i],modify(st[x],b[x]),have[x]=b[x]=0; cn=0; } read(x),x=fin[x],printf("%d\n",sum(en[x])-sum(st[x]-1)); } } return 0; }