码迷,mamicode.com
首页 > 编程语言 > 详细

[BZOJ2434]NOI2011阿狸的打字机|AC自动机|fail树|树状数组

时间:2015-04-20 09:36:29      阅读:455      评论:0      收藏:0      [点我收藏+]

标签:ac自动机   树状数组   

这题真是太神了,好多实用的技巧。。首先肯定是要先把每个要输出的串当模式串把自动机给建出来的,如果一个一个串复制出来再一个个插入显然非常慢。。。我们用在自动机上插入模式串的方法来建,初始时在0,新加一个字符就想下爬(或者新建),维护一个父亲指针,删除的时候就可以爬上去,这样就可以O(n)建出来了。。

再考虑询问的问题,每次把串拿出来再放进自动机跑一遍显然太慢。。这里需要用到一个叫做fail树的东西,就是把fail指针当做边建成的一颗树。。比如fail(i)=j,那么i在fail树上的父节点就是j。。对于每个询问,若x串在y串中出现了,那么root到y串节点的路径上必有一点的next指针是x串节点,放到fail树上看,询问(x,y)的答案就是x在fail上的子树中在root到y串节点的路径上的节点的数量,看到子树和,考虑用dfs序来解决。。把所有询问像链表一样挂到各自的y串节点上,然后模拟一开始的插入过程,每向下爬到一个节点就将它在dfs序中的左边界+1,每向上离开一个节点就将它在dfs序中的左边界-1,每当爬到一个danger节点就把所有这个节点的询问解决,用树状数组维护前缀和即可。。

#include<iostream>
#include<cstdio>
#include<memory.h>
#define N 100005
using namespace std;
char s[N];
struct edge{
	int e,next;
}ed[N];
struct Query{
	int e,next,xu;
}q[N];
int i,j,ne=0,x,y,t=0,nq=0,n,m,a[N][27],head[N],que[N],next[N],fa[N],l[N],r[N],c[N*2],g[N],ans[N],pos[N];//cnt为串数,nd为AC自动机点数,ne为fail树边数,nq为询问链表边数 
void inse(int s,int e)
{
	ed[++ne].e=e;
	ed[ne].next=g[s];g[s]=ne;
}
void insq(int x,int y,int i)
{
	q[++nq].e=x;q[nq].xu=i;
	q[nq].next=head[y];head[y]=nq;
}
int lowbit(int x){return x&-x;}
void add(int x,int k)
{
	for (int i=x;i<=t;i+=lowbit(i)) c[i]+=k;
}
int sum(int x)
{
	int ans=0;
	for (int i=x;i;i-=lowbit(i)) ans+=c[i];
	return ans;
}
void init()
{
	int i,now=1,cnt=0,nd=1;
	memset(a,0,sizeof(a));memset(c,0,sizeof(c));
	memset(head,0,sizeof(head));memset(g,0,sizeof(g));
	n=strlen(s);fa[1]=0;
	for (i=0;i<26;i++) a[0][i]=1;
	for (i=0;i<n;i++)
		if (s[i]=='P') pos[++cnt]=now;
		else if (s[i]=='B') now=fa[now];
		else
		{
			if (!a[now][s[i]-'a'])
			{
				a[now][s[i]-'a']=++nd;
				fa[nd]=now;
			}
			now=a[now][s[i]-'a'];
		}
}
void build()
{
	next[1]=0;inse(0,1);
	int head=1,tail=1,get,k,i;
	que[1]=1;
	while (head<=tail)
	{
		get=que[head++];
		for (i=0;i<26;i++)
			if (a[get][i])
			{
				k=next[get];
				while (!a[k][i]) k=next[k];k=a[k][i];
				next[a[get][i]]=k;inse(k,a[get][i]);
				que[++tail]=a[get][i];
			}
	}
}
void dfs(int x)
{
	l[x]=++t;
	for (int j=g[x];j;j=ed[j].next)
		dfs(ed[j].e);
	r[x]=++t;
}
void solve()
{
	int now=1,cnt=0,i,j;
	add(l[1],1);
	for (i=0;i<n;i++)
		if (s[i]=='P')
		{
			cnt++;
			for (j=head[now];j;j=q[j].next)
				ans[q[j].xu]=sum(r[q[j].e])-sum(l[q[j].e]-1);
		}
		else if (s[i]=='B') add(l[now],-1),now=fa[now];
		else now=a[now][s[i]-'a'],add(l[now],1);
}
int main()
{
	freopen("2434.in","r",stdin);
	scanf("%s",s);
	scanf("%d",&m);
	init();build();dfs(0);
	for (i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		insq(pos[x],pos[y],i);
	}
	solve();
	for (i=1;i<=m;i++) printf("%d\n",ans[i]);
}


[BZOJ2434]NOI2011阿狸的打字机|AC自动机|fail树|树状数组

标签:ac自动机   树状数组   

原文地址:http://blog.csdn.net/tag_king/article/details/45132273

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!