bzoj2434: [Noi2011]阿狸的打字机

传送门:http://www.lydsy.com:808/JudgeOnline/problem.php?id=2434

一个讲得很详细的题解:http://blog.csdn.net/huzecong/article/details/7769988

思路:这题的想法有点神啊....

先构建AC自动机,然后怎么判断一个串b是a的子串呢?用fail指针就可以了。如果a串中有节点可以通过fail指针走到b的终止节点,那么b就在a中出现过。有n个节点可以走到b,那么b就出现过n次。

现在就有一个暴力的想法,枚举a串的每个节点的fail看是否能到b,但是这是显然会T的。

然后我们可以倒过来想,把fail指针反向,建一棵fail树,对于b串,统计子树中有多少个a串的节点即可。

子树的节点的dfs序是相连的。

这样我们就可以用树状数组维护一下就好了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
const int maxn=100010;
using namespace std;
int m,len,num,preq[maxn],nowq[maxn],sonq[maxn],l[maxn],r[maxn],bit[maxn],pos[maxn],cnt,pre[maxn],now[maxn],son[maxn],ans[maxn];char s[maxn],ch;
void change(int x,int val){for (;x<=cnt;x+=x&-x) bit[x]+=val;}
int query(int x){
	int res=0;
	for (;x;x-=x&-x) res+=bit[x];
	return res;
}
void add(int a,int b){pre[++num]=now[a],now[a]=num,son[num]=b;}
void read(int &x){
	for (ch=getchar();!isdigit(ch);ch=getchar());
	for (x=0;isdigit(ch);ch=getchar()) x=x*10+ch-'0';
}

struct AC_DFA{
	int tot,ch[maxn][26],fail[maxn],q[maxn],fa[maxn],head,tail;
	void build(){
		int p=0,id=0;
		for (int i=0;i<len;i++)
			if (s[i]=='P') pos[++id]=p;
			else if (s[i]=='B') p=fa[p];
			else{
				if (!ch[p][s[i]-'a']) ch[p][s[i]-'a']=++tot,fa[tot]=p;
				p=ch[p][s[i]-'a'];
			}
	}
	void getfail(){
		head=0,q[tail=1]=0,fail[0]=-1;
		while (head!=tail){
			int x=q[++head];
			for (int i=0;i<26;i++)
				if (ch[x][i])
					q[++tail]=ch[x][i],fail[ch[x][i]]=x==0?0:ch[fail[x]][i];
				else ch[x][i]=x==0?0:ch[fail[x]][i];
		}
	}
	void dfs(int x){
		l[x]=++cnt;
		for (int y=now[x];y;y=pre[y])
			dfs(son[y]);
		r[x]=cnt;
	}
	void work(){
		int p=0,id=0;
		change(l[0],1);
		for (int i=0;i<len;i++)
			if (s[i]=='P'){
				id++;
				for (int y=nowq[id];y;y=preq[y]){
					int t=pos[sonq[y]];
					ans[y]=query(r[t])-query(l[t]-1);
				}
			}
			else if (s[i]=='B') change(l[p],-1),p=fa[p];
			else p=ch[p][s[i]-'a'],change(l[p],1);
	}
}T;

int main(){
	scanf("%s%d",s,&m);len=strlen(s);
	for (int i=1,x,y;i<=m;i++){
		read(x),read(y);
		preq[i]=nowq[y],sonq[i]=x,nowq[y]=i;
	}
	T.build(),T.getfail();
	for (int i=1;i<=T.tot;i++) add(T.fail[i],i);
	T.dfs(0),T.work();
	for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
	return 0;
}


posted @ 2015-07-26 17:50  orzpps  阅读(223)  评论(0编辑  收藏  举报