luogu P5840 [COCI2015]Divljak
题面传送门
真是一道毒瘤题。
容易想到先把所有\(S\)插入AC自动机,然后在每插入一个\(T\)时算贡献。
但是有一个问题是插入\(T\)时可能会对一个串造成许多贡献。
这时就有一个很妙的trick:把所有待插入位置按dfs序排序,然后把每个点单点加的时候在每两个点lca处差分,这样就可以保证每个串只被计算一次贡献。
然后把fail树搞出来,在树状数组维护单点加区间查即可。
听到这里您可能会觉得这道题太水了,以至于您能爆切它。
然而这道题有如下几个坑点:
这道题空间看上去很够,实际如果用倍增求lca就一点不够。一定要用树剖求lca
题面中说的所有字符串总长仅仅指\(S\)总长,\(T\)不计算在内。
如果你用线段树那么恭喜你被卡常了。
大概就这么点,时间复杂度\(O(\sum{|S|}+qlogn)\)
代码实现:
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#define beg(x) int cur=s.h[x]
#define end cur
#define go cur=tmp.z
using namespace std;
int n,m,k,x,y,z,g[2000039],now,dh,bg[2000039],cnt,id[400039],st[2000039],sh,lcas;
char a[2000039];
int siz[2000039],top[2000039],son[2000039],d[2000039],fa[2000039];
struct yyy{int to,z;};
struct ljb{
int head,h[2000039];yyy f[2000039];
inline void add(int x,int y){f[++head]=(yyy){y,h[x]};h[x]=head;}
}s;
struct AC{int son[26],fail;}f[2000039];
inline void get(int x){
register int i;now=0;
for(i=1;i<=k;i++) now=f[now].son[a[i]-'a']?f[now].son[a[i]-'a']:(f[now].son[a[i]-'a']=++cnt);id[x]=now;
}
queue<int> q;
inline void bfs(){
register int i;
for(i=0;i<=25;i++) if(f[0].son[i]) q.push(f[0].son[i]);
while(!q.empty()){
now=q.front();q.pop();
for(i=0;i<=25;i++){
if(f[now].son[i]) f[f[now].son[i]].fail=f[f[now].fail].son[i],q.push(f[now].son[i]);
else f[now].son[i]=f[f[now].fail].son[i];
}
}
}
inline void dfs1(int x,int last){
fa[x]=last;siz[x]=1;d[x]=d[last]+1;yyy tmp;
for(beg(x);end;go)tmp=s.f[cur],dfs1(tmp.to,x),siz[x]+=siz[tmp.to],son[x]=siz[son[x]]>siz[tmp.to]?son[x]:tmp.to;
}
inline void dfs2(int x,int last){
top[x]=last;bg[x]=++dh;if(!son[x]) return;yyy tmp;dfs2(son[x],last);
for(beg(x);end;go) {
tmp=s.f[cur];
if(tmp.to^son[x]) dfs2(tmp.to,tmp.to);
}
}
inline int lca(int x,int y){
while(top[x]!=top[y])d[top[x]]>d[top[y]]?(x=fa[top[x]]):(y=fa[top[y]]);
return d[x]>d[y]?y:x;
}
inline bool cmp(int x,int y){return bg[x]<bg[y];}
inline void get(int x,int y){while(x<=dh)g[x]+=y,x+=x&-x;}
inline int find(int x){int ans=0;while(x)ans+=g[x],x-=x&-x;return ans;}
int main(){
freopen("1.in","r",stdin);
register int i,j;
scanf("%d",&n);
for(i=1;i<=n;i++)scanf("%s",a+1),k=strlen(a+1),get(i),id[i]++;bfs();
for(i=1;i<=cnt;i++) s.add(f[i].fail+1,i+1);dfs1(1,0);dfs2(1,1);
scanf("%d",&m);
for(i=1;i<=m;i++){
scanf("%d",&x);
if(x==1){
scanf("%s",a+1);k=strlen(a+1);now=sh=0;
for(j=1;j<=k;j++)now=f[now].son[a[j]-'a'],st[++sh]=now+1;
sort(st+1,st+sh+1,cmp);
for(j=1;j<=sh;j++){
if(j>1)lcas=lca(st[j],st[j-1]),get(bg[lcas],-1);
get(bg[st[j]],1);
}
}
else{
scanf("%d",&x);x=id[x];printf("%d\n",find(bg[x]+siz[x]-1)-find(bg[x]-1));
}
}
}