这题真是太神了,好多实用的技巧。。首先肯定是要先把每个要输出的串当模式串把自动机给建出来的,如果一个一个串复制出来再一个个插入显然非常慢。。。我们用在自动机上插入模式串的方法来建,初始时在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树|树状数组
原文地址:http://blog.csdn.net/tag_king/article/details/45132273