[AC自动机 fail树 树链的并] BZOJ 3881 [Coci2015]Divljak

建出AC自动机和fail树

然后每一个串在自动机上跑一遍 然后就是对应的节点到fail树根的路径的并都加一

路径的并有一个很好地idea 

可以用两种dfs序 入栈序和出栈序

入栈序小于等于子树的last 减去 出栈序小于自己的 就是到根的路径

#include<cstdio>
#include<cstdlib>
#include<algorithm>
//#define dprintf(...) fprintf(stderr,__VA_ARGS__)
using namespace std;

inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x)
{
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

inline int read(char *s)
{
	char c=nc(); int len=0;
	for (;!(c>='a' && c<='z');c=nc());
	for (;c>='a' && c<='z';s[++len]=c,c=nc()); s[++len]=0; return len-1;
}

const int N=2000005,K=20;

struct BIT{
	int maxn,c[N];
	inline void init(int n){ 
		maxn=n; 
	}
	inline void add(int x,int r){ 
		for (int i=x;i<=maxn;i+=i&-i) c[i]+=r;
	}
	inline void add(int s,int t,int r){
		if (s>t) return; add(s,r); add(t+1,-r);
	}
	inline int val(int x){
		int ret=0; for (int i=x;i;i-=i&-i) ret+=c[i]; return ret;
	}
}bit1,bit2;

#define V G[p].v
struct edge{
	int u,v,next;
}G[N];
int head[N],inum;
inline void add(int u,int v,int p){
	G[p].u=u; G[p].v=v; G[p].next=head[u]; head[u]=p;
}

int clk1,clk2,pre[N],size[N],last[N],depth[N],fat[N][K];

inline int LCA(int u,int v){
	if (depth[u]<depth[v]) swap(u,v);
	for (int k=K-1;~k;k--)
		if ((depth[u]-depth[v])>>k&1)
			u=fat[u][k];
	if (u==v) return u;
	for (int k=K-1;~k;k--)
		if (fat[u][k]!=fat[v][k])
			u=fat[u][k],v=fat[v][k];
	return fat[u][0];
}

inline void dfs(int u,int fa){
	pre[u]=++clk1; size[u]=1; fa==-1?depth[u]=1:depth[u]=depth[fa]+1;
	fat[u][0]=fa;
	for (int k=1;k<K;k++)
		fat[u][k]=(fat[u][k-1]==-1)?-1:fat[fat[u][k-1]][k-1];
	for (int p=head[u];p;p=G[p].next)
		dfs(V,u),size[u]+=size[V];
	last[u]=++clk2;
}

int ncnt,root,ch[N][26];
int back[N];
inline int Insert(char *s,int len){
	int p=root;
	for (int i=1;i<=len;i++){
		if (!ch[p][s[i]-'a']) ch[p][s[i]-'a']=++ncnt;
		p=ch[p][s[i]-'a'];
	}
	return p;
}
int Q[N],l,r;
int f[N];
inline void Build(){
	f[0]=0; l=r=-1; int u,v;
	for (int i=0;i<26;i++)
		if (ch[0][i])
			Q[++r]=ch[0][i],f[ch[0][i]]=0;
	while (l<r)
	{
		u=Q[++l];
		for (int i=0;i<26;i++)
		{
			v=ch[u][i];
			if (!v) { ch[u][i]=ch[f[u]][i]; continue; }
			Q[++r]=v;
			f[v]=ch[f[u]][i];
		}
	}
	for (int i=1;i<=ncnt;i++)
		add(f[i],i,++inum);
	dfs(0,-1); bit1.init(clk1); bit2.init(clk2);
}

inline void Add(int u,int r){
	int t=pre[u]+size[u]-1;
	bit1.add(1,t,r);
	bit2.add(1,last[u]-1,-r);
}
inline int Query(int u){
	return bit1.val(pre[u])+bit2.val(last[u]);
}

char str[N]; int Len;
int lst[N],pnt;

inline bool cmp(int a,int b){
	return pre[a]<pre[b];
}

inline void Modify(){
	int p=root; pnt=0;
	for (int i=1;i<=Len;i++)
	{
		p=ch[p][str[i]-'a'];
		lst[++pnt]=p;
	}
	sort(lst+1,lst+pnt+1,cmp);
	pnt=unique(lst+1,lst+pnt+1)-lst-1;
	Add(lst[1],1);
	for (int i=2;i<=pnt;i++)
	{
		Add(lst[i],1);
		int lca=LCA(lst[i],lst[i-1]);
		Add(lca,-1);
	}
//	for (int i=0;i<=ncnt;i++) dprintf("%d ",Query(i)); dprintf("\n");
}

int n;

int main()
{
	int Q,order,x;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	read(n);
	for (int i=1;i<=n;i++)
		Len=read(str),back[i]=Insert(str,Len);
	Build();
	read(Q);
	while (Q--){
		read(order);
		if (order==1)
			Len=read(str),Modify();
		else
			read(x),printf("%d\n",Query(back[x]));
	}
	return 0;
}