BZOJ3881 [Coci2015]Divljak 【AC自动机+树上差分】

dark

SOL

• 类似于阿狸的打字机？ 只是这道题需要一个路径并。

• 具体的，对Alice的串建出ac自动机，每次拿新加入的$Bob$串在上面跑匹配，把所有经过点拿出来，在$fail$树上给到根的路径打$+1$标记。但是多次出现算作一次，所以做一个路径并就可以了。即给点按照$fail$树上的$dfs$序排序，在相邻两个点的$lca$处减掉1。

• 对于询问把单点查值用树上数组转化成子树求和。

• 其实这些都可以用广义$sam$和$par$树实现。只是大材小用了，ac自动机即可。

这题卡欧拉序版lca，要用树剖求。。

---

CODE

#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define pf printf
#define ll long long
#define cs const
#define db double
#define ri register int
#define gc getchar()
#define in red()
inline int red(){
	int num=0,f=1;char c=gc;
	for(;!isdigit(c);c=gc)if(c=='-')f=-1;
	for(;isdigit(c);c=gc)num=(num<<1)+(num<<3)+(c^48);
	return num*f;
}
cs int N=2e6+10;
int fa[N],ch[N][26],pos[N],dfn[N],tot=1,tim=0,siz[N];
inline bool cmp(int a,int b){return dfn[a] < dfn[b];}
vector<int>V,g[N];
namespace ST{
	int top[N],dep[N],son[N];
	void dfs(int u,int f){
		dep[u]=dep[fa[u]]+1;top[u]=f;
		for(ri i=g[u].size()-1;i>=0;--i){
			if(siz[g[u][i]]>siz[son[u]])son[u]=g[u][i];
		}
		if(son[u])dfs(son[u],f);
		for(ri i=g[u].size()-1;i>=0;--i){
			if(g[u][i]^son[u])dfs(g[u][i],g[u][i]);
		}
	}
	inline int Lca(int x,int y){
		while(top[x]^top[y]){
			if(dep[top[x]]<dep[top[y]])swap(x,y);
			x=fa[top[x]];
		}
		return dep[x]<dep[y] ? x : y;
	}
}
inline void dfs(int u){
	dfn[u]=++tim;siz[u]=1;
	for(ri i=g[u].size()-1;i>=0;--i)dfs(g[u][i]),siz[u]+=siz[g[u][i]];
}
#define lb(x) (x&(-x))
int sum[N];
inline void upt(int p,int k){for(;p<=tot;p+=lb(p))sum[p]+=k;}
inline int qy(int p){int ans=0;for(;p>=1;p-=lb(p))ans+=sum[p];return ans;}
inline void insert(char *s,int &t){
	for(ri i=0,up=strlen(s),now=1;i<up;++i,t=now){
		int c=s[i]-'a';
		now=ch[now][c] ? ch[now][c] : ch[now][c]=++tot;
	}
}
inline void bfs(){
	queue<int> q;
	for(ri i=0;i<26;++i){
		if(ch[1][i])fa[ch[1][i]]=1,q.push(ch[1][i]),g[1].push_back(ch[1][i]);
		else ch[1][i]=1;
	}
	while(!q.empty()){
		int u=q.front();q.pop();
		for(ri i=0;i<26;++i){
			if(ch[u][i])fa[ch[u][i]]=ch[fa[u]][i],g[fa[ch[u][i]]].push_back(ch[u][i]),q.push(ch[u][i]);
			else ch[u][i]=ch[fa[u]][i];
		}
	}
}
inline void add(char *s){
	V.clear();
	for(ri i=0,up=strlen(s),now=1;i<up;++i){
		now=ch[now][s[i]-'a'];
		V.push_back(now);
	}
	V.push_back(1);
	sort(V.begin(),V.end(),cmp);	
	for(ri i=1,up=V.size();i<up;++i){
		int L=ST::Lca(V[i],V[i-1]);
		upt(dfn[L],-1);upt(dfn[V[i]],1);
	}
}
int n,m;
char s[N];
signed main(){
//	freopen("data.in","r",stdin);
	n=in;
	for(ri i=1;i<=n;++i){
		sf("%s",s);
		insert(s,pos[i]);
	}
	bfs();
	dfs(1);	
	ST::dfs(1,1);
	m=in;
	while(m--){
		int op=in,x;
		if(op==1)sf("%s",s),add(s);
		if(op==2){
			x=in;
			cout<<qy(dfn[pos[x]]+siz[pos[x]]-1)-qy(dfn[pos[x]]-1)<<'\n';	
		}
	}



	return 0;
}