CF163E. e-Government AC自动机+fail树+分块

原题：http://codeforces.com/contest/163/problem/E

题解：给k个字符串，维护3种操作，添加一个字符串，删除一个字符串，查询模式串在主串中出现了几次。暴力来做，跑ac自动机，统计所有的失败指针对应的字符串。考虑如何优化，将失败指针反向，那么模式串的子树就是答案。用深搜序转换成线性，再用对应的数据结构维护就可以，可以用线段树，树状数组，数列分块等。我用的数列分块。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
const int N=1e6+10;
const int M=1e3+10;
const int Z=26;
struct E{int x,y,next;}mm[N];
int ch[N][Z],nxt[N],h[N],l[N],r[N],a[N],sum[M],pos[N];
int tot,n,k,cnt,num,m,end1[N];
bool vis[N],v[N];
char s[N];
inline void ins(char *s,int ind){
	int len=strlen(s+1);int p=1;
	for(int i=1;i<=len;i++){
		int c=s[i]-'a';
		if(!ch[p][c]) ch[p][c]=++tot;
		p=ch[p][c];
	}
	end1[ind]=p;
}
void bfs(){
	queue<int> q;
	for(int i=0;i<Z;i++) ch[0][i]=1;
	nxt[1]=0;q.push(1);
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=0;i<Z;i++)
			if(!ch[u][i]) ch[u][i]=ch[nxt[u]][i];
			else {
				q.push(ch[u][i]);int v;
				for(v=nxt[u];v>0 && !ch[v][i];v=nxt[v]);
				nxt[ch[u][i]]=ch[v][i];
			}	
	}
}
inline void insE(int x,int y){
	mm[++cnt].x=x;mm[cnt].y=y;mm[cnt].next=h[x];h[x]=cnt;
}
void dfs(int u){
	l[u]=++num;vis[u]=1;
	for(int k=h[u];k;k=mm[k].next){
		int v=mm[k].y;if(!vis[v]) dfs(v);
	}
	r[u]=num;
}
void add(int l,int r,int c){
	for(int i=l;i<=min(r,pos[l]*m);i++) a[i]+=c;
	if(pos[l] != pos[r]){
		for(int i=(pos[r]-1)*m+1;i<=r;i++) a[i]+=c;
	}
	for(int i=pos[l]+1;i<=pos[r]-1;i++) sum[i]+=c;
}
inline int query(int x){return a[x]+sum[pos[x]];}
int main(){
//	freopen("cf163e.in","r",stdin);
	tot=1;memset(h,0,sizeof h);num=0;cnt=0;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=k;i++) scanf("%s",s+1),ins(s,i);
	bfs();
	for(int i=2;i<=tot;i++) insE(nxt[i],i);
	dfs(1);
	m=sqrt(num);
	for(int i=1;i<=num;i++) pos[i]=(i-1)/m+1;
	for(int i=1;i<=k;i++) add(l[end1[i]],r[end1[i]],1),v[i]=1;	
	while(n--){
		scanf("%s",s);int len=strlen(s)-1;
		if( s[0]== '+' ){
			int x=0;
			for(int j=1;j<=len;j++) x=(x<<1)+(x<<3)+s[j]-'0';
			if(!v[x])add(l[end1[x]],r[end1[x]],1),v[x]=1;
		}
		
		if( s[0]== '-' ){
			int x=0;
			for(int j=1;j<=len;j++) x=(x<<1)+(x<<3)+s[j]-'0';
			if(v[x])add(l[end1[x]], r[end1[x]],-1),v[x]=0;
		}
		if( s[0]== '?' ){
			int p=1;int ans=0;
			for(int i=1;i<=len;i++){
				int c=s[i]-'a';int k=ch[p][c];
				p=ch[p][c];
				ans+=query(l[k]);
			}
			printf("%d\n",ans);	 
		}
	}
	return 0;
}