【AC自动机复习】 noi2011阿狸的打字机

       有些算法学了很久没有复习，今天重刷了两道后缀数组的题，然后想找到AC自动机的题目，毕竟以前也没写过几遍。

       然后尴尬的发现去年noi的阿狸的打字机还没写，去年暑假因为不懂AC自动机，没有去动它，后来就一直忘记了=。=！

 

       不说废话了。

 

       将所有输出的单词存入AC自动机，由于题目特殊，建起来比一般的AC自动机要简单。

       考虑打裸，对于询问（i，j），枚举j单词上的每一个点，如果该点可以通过fail指针指向i的末尾，那么ans++；

        这样当然是不能AC的，注意到，如果把fail指针反向，然后建立一颗fail树，那么，从j上的点可以通过fail指针走到i的数量，等价于在fail tree上，以i根的子树中，

有多少个j单词上的点。

       这样，就不难想到，如果我们假设fail tree 上，之后j单词上的点有权值，那么我们可以轻易的用dfs序列+树状数组/线段树统计出来。那么，可以考虑dfs这个trie， dfs过程中修改fail tree 上的权值，很容易可以做到dfs 到j单词末尾是，仅有j单词上的点有权值 ，这样用离线的方法可以处理所有询问。

 

      AC自动机貌似不会怎么出错，一遍过了。

   

# include <cstdlib>
# include <cstdio>
# include <cmath>
# include <cstring>

using namespace std;

const int maxn = 100000+5;
struct list
{
	int top, linke[maxn*2+10], sum[maxn*2+10], next[maxn*2+10];
	void  link(int x, int y)
	{
		++top; next[top]= linke[x]; linke[x]=top; sum[top]=y;
	}	
}ans, edge;

int n, len;
int tot,end[maxn], fail[maxn], map[maxn][26], pre[maxn];
char s[maxn];
int l,r,que[maxn],st[maxn],have,ask[maxn], sol[maxn];
int head[maxn], tail[maxn], go[maxn], c[maxn];

void read()
{
	int i,x,y;
	scanf("%s\n", s+1);
	scanf("%d", &n);
	for (i = 1; i <= n; i++)
	{ 
	  scanf("%d%d", &x, &y);
	  ans.link(y, x);
	}	  
}

void build_AC()
{
	int i,now=1; tot=1;len= strlen(s+1);
	for (i = 1; i <= len; i++)
	{
		if (s[i]=='P') end[now]=++have, ask[have]=now; 
		else if (s[i]=='B') now=pre[now];
		else now= map[now][s[i]-'a']==0? pre[++tot]=now,map[now][s[i]-'a']=tot: map[now][s[i]-'a'];
	}
	for (i = 0; i <= 25; i++)
	  if (map[1][i]) que[++r]=map[1][i], fail[que[r]]=1, edge.link(1, que[r]);
	for (l=1;l<= r;l++)
	  for (i=0; i<=25; i++)
		if (map[que[l]][i])
	    {
			que[++r]=map[que[l]][i];
			for (now=fail[que[l]]; now!=1&& map[now][i]==0;)now=fail[now];
			fail[que[r]]= map[now][i]?map[now][i]:now;
			edge.link(fail[que[r]], que[r]);
		}
}

void pdfs(int now)
{
	st[++st[0]]=now; head[now]=st[0]; go[now]=st[0];
	for (int ke= edge.linke[now]; ke!=0; ke=edge.next[ke])
	   pdfs(edge.sum[ke]);
	tail[now]=st[0];
}

void modify(int x, int d)
{
	for (;x<=st[0];x+= x&(-x))
	  c[x]+= d;
}
int query(int x)
{
	int ask=0;
	for (;x>0;x-= x&(-x))
	  ask+= c[x];
	return ask;
}
void work()
{
	int i,now=1,ks;
    for (i=1; i <= len; i++)
    {
		if (s[i]=='P');
		else if (s[i]=='B') modify(go[now], -1),now=pre[now];
		else now= map[now][s[i]-'a'], modify(go[now], 1);
		if (ks=end[now]) 
		{
		  for (int ke=ans.linke[ks]; ke!= 0; ke=ans.next[ke])
		    sol[ke]= query(tail[ask[ans.sum[ke]]])- query(head[ask[ans.sum[ke]]]-1);
		  end[now]=0;
		};
	}
}

int main()
{
	//freopen("type.in", "r", stdin);
	//freopen("type.out", "w", stdout);
	read(); 
	build_AC();
	pdfs(1);
	work();
	for (int i = 1; i <= n; i++) printf("%d\n", sol[i]);
	return 0;
}