【NOI2011】阿狸的打字机【AC自动机】【离线】【树状数组】

好题啊好题

这道题题意很清楚，做起来很毒瘤。

首先我们应该可以看出这是一个多模式串多文本串匹配问题，所以我们往自动机上面想。

支持加入点，删点，记录打印标记，最后查询答案。

删点无非就是几个father，然后回去。打印标记就是记录这次打印在AC自动机的哪个节点发生的。

然后给每个自动机节点记录有哪些打印是发生在这个结点的。我开了个vector来记。

因为问题很多，一个一个做会出事，所以我使用了离线算法。对于一个询问x在y里出现多少次。我们按y为关键字sort，当在AC自动机上走到y次打印对应的结点时，所有可能出现在其内的子串都应该在它的fail树上出现过。直接统计fail树的子树size就能得到这个x出现了几次。

单点加减，子树求和，我使用dfs序+树状数组来完成这个维护。

如此甚好。

#include<bits/stdc++.h>
using namespace std;
#define in read()
#define int long long
int in{
	int cnt=0,f=1;char ch=0;
	while(!isdigit(ch)){
		ch=getchar();if(ch=='-')f=-1;
	}
	while(isdigit(ch)){
		cnt=cnt*10+ch-48;
		ch=getchar();
	}return cnt*f;
}
struct node{
	int ch[26],fail;
	int vis[26];
	vector<int> que;int fa;
}t[200003];
int cnt;
struct bili{
	int x,y,id;
}que[100003];
int n,m;
char ch[233333];
int T[100003],dfn[100003],low[100003],id,first[100003],nxt[100003],to[100003],tot;
int cntque;int ans[100003];int ql[100003];int qr[100003];int ed[100003];
void add(int a,int b){
	nxt[++tot]=first[a];first[a]=tot;to[tot]=b; 
}
bool cm(bili a,bili b){
	return a.y<b.y;
}
void getfail(){
	queue<int> q;
	for(int i=0;i<26;i++){if(t[0].ch[i])q.push(t[0].ch[i]);}
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=0;i<26;i++){
			if(t[u].ch[i])t[t[u].ch[i]].fail=t[t[u].fail].ch[i],q.push(t[u].ch[i]);
			else t[u].ch[i]=t[t[u].fail].ch[i];
		}
	}
}
void dfs(int u){
	//cout<<"dfs1  "<<u<<endl;
	dfn[u]=++id;
	for(int i=first[u];i;i=nxt[i]){
		int v=to[i];dfs(v);
	}low[u]=id;
}
int lowbit(int x){
	return x&(-x);
}
void modify(int x,int key){
	while(x<=id){
		T[x]+=key;x+=lowbit(x);
	}
}
int num[100003],cntnum;
int query(int x){
	int sum=0;
	while(x){
		sum+=T[x];x-=lowbit(x);
	}return sum;
}
void dfs2(int u){
	//cout<<"dfs2  "<<u<<" "<<" dfn "<<dfn[u]<<endl;
	modify(dfn[u],1);
	//for(int i=1;i<=cnt;i++)cout<<query(i)<<" ";cout<<endl;
	if(t[u].que.size()){//cout<<"  get into"<<endl;
		cntnum=0;
		sort(t[u].que.begin(),t[u].que.end());
		
		for(int i=0;i<t[u].que.size();i++){
			if(i==0)num[++cntnum]=t[u].que[i];
			else{
				if(t[u].que[i]!=t[u].que[i-1])num[++cntnum]=t[u].que[i];
			}
		}
	//	cout<<"question "<<" ";for(int i=1;i<=cntnum;i++)cout<<num[i]<<" ";cout<<endl;
		for(int i=1;i<=cntnum;i++){//cout<<"id ";
			for(int j=ql[num[i]];j<=qr[num[i]];j++){if(!j)continue;
				//cout<<que[j].id<<" ";cout<<" x "<<que[j].x<<" ";cout<<" ed "<<ed[que[j].x]<<" ";
				//cout<<" low "<<low[ed[que[j].x]]<<" "<<" dfn "<<dfn[ed[que[j].x]];
				ans[que[j].id]=query(low[ed[que[j].x]])-query(dfn[ed[que[j].x]]-1);
			}//cout<<endl;
		}
	}
	for(int i=0;i<26;i++){
		if(t[u].vis[i])dfs2(t[u].vis[i]);
	}
	modify(dfn[u],-1);
}
signed main(){
	scanf("%s",ch+1);int len=strlen(ch+1);
	int rt=0;
	for(int i=1;i<=len;i++){
		if(ch[i]>='a'&&ch[i]<='z'){
			int x=ch[i]-'a';
			if(!t[rt].ch[x]){
				t[rt].ch[x]=++cnt;t[cnt].fa=rt;
			}rt=t[rt].ch[x];
		}else if(ch[i]=='P')ed[++cntque]=rt,t[rt].que.push_back(cntque);
		else rt=t[rt].fa;
	}
	for(int i=0;i<=cnt;i++)for(int j=0;j<26;j++)t[i].vis[j]=t[i].ch[j];
	m=in;getfail();for(int i=1;i<=cnt;i++)add(t[i].fail,i);dfs(0);//for(int i=1;i<=cnt;i++)cout<<t[i].fail<<" ";cout<<endl;
	for(int i=1;i<=m;i++)que[i].x=in,que[i].y=in,que[i].id=i;
	sort(que+1,que+m+1,cm);
	for(int i=1,pos=1;i<=m;i=pos){
		ql[que[i].y]=pos;
		while(pos<=m&&que[pos].y==que[i].y)++pos;
		qr[que[i].y]=pos-1;
	}
	dfs2(0);
	for(int i=1;i<=m;i++)cout<<ans[i]<<'\n';
	return 0;
}