【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;
}