BZOJ 2434 阿狸的打字机 补全AC自动机 ( Trie图 ) fail树 树状数组 DFS序列

大家都很强， 可与之共勉 。

2434: [Noi2011]阿狸的打字机

Time Limit: 10 Sec Memory Limit: 256 MB
[Submit][Status][Discuss]
Description

阿狸喜欢收藏各种稀奇古怪的东西，最近他淘到一台老式的打字机。打字机上只有28个按键，分别印有26个小写英文字母和’B’、’P’两个字母。

经阿狸研究发现，这个打字机是这样工作的：

l 输入小写字母，打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。

l 按一下印有’B’的按键，打字机凹槽中最后一个字母会消失。

l 按一下印有’P’的按键，打字机会在纸上打印出凹槽中现有的所有字母并换行，但凹槽中的字母不会消失。

例如，阿狸输入aPaPBbP，纸上被打印的字符如下：

a

aa

ab

我们把纸上打印出来的字符串从1开始顺序编号，一直到n。打字机有一个非常有趣的功能，在打字机中暗藏一个带数字的小键盘，在小键盘上输入两个数(x,y)（其中1≤x,y≤n），打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。

阿狸发现了这个功能以后很兴奋，他想写个程序完成同样的功能，你能帮助他么？

Input

输入的第一行包含一个字符串，按阿狸的输入顺序给出所有阿狸输入的字符。

第二行包含一个整数m，表示询问个数。

接下来m行描述所有由小键盘输入的询问。其中第i行包含两个整数x, y，表示第i个询问为(x, y)。

Output

输出m行，其中第i行包含一个整数，表示第i个询问的答案。

Sample Input

aPaPBbP
3
1 2
1 3
2 3

Sample Output

2
1
0

HINT

1<=N<=10^5
1<=M<=10^5

输入总长<=10^5
Source

Trie

[Submit][Status][Discuss]

第一次写数组版的AC自动机Qwq， 主要是不想hash指针Qwq（不好建图hhh） 。

题解 :
这是一道神题QQQ
大致题意如下给出QQQ个串，求第x个串在第y个串中间出现了多少次 。
大不了咱们跑Kmp**吼吼吼**
去TM的BZOJ没有部分分评测hhh 。
无奈正解 ：
直接说正解吧qwq

考虑建一个补全自动机 （Trie图），然后得到fail数组。然后建出fail树（即fail [i]向i连边），那么只要某个串的节点的fail指到x的尾节点，那么x一定在该串中出现。所以我们对一个节点打标记，查询时找x的子树中有多少个标记就可以了。

用Dfs序 + 树状数组

完美解决问题~
最后一次用蜜汁DEV C++手动再见（不让我用调试）

AC代码 ：

/**************************************************************    Problem: 2434    User: Lazer2001    Language: C++    Result: Accepted    Time:524 ms    Memory:19508 kb****************************************************************/# include <bits/stdc++.h>const int N = 100005 ;int go [N] [26], fail [N], fa [N], ncnt ;int pos [N] ;char buf [N] ;struct edge  {    int to, nxt ;} g [N << 1] ;int head [N], ecnt ;inline void add_edge ( int u, int v )  {    g [++ ecnt] = ( edge )  {  v, head [u]  } ;  head [u] = ecnt ;}struct Q {    int x, qid ;} ;std :: vector < Q > qry [N] ;void Extend ( char* s )  {    int cur ( 0 ), id ( 0 ) ;    for ( char* pt = s ; *pt ; ++ pt )  {        if ( *pt == 'P' )  {            pos [++ id] = cur ;            continue ;        }        if ( *pt == 'B' )  {            cur = fa [cur] ;            continue ;        }        int k = *pt - 'a' ;        if ( ! go [cur] [k] )   go [cur] [k] = ++ ncnt, fa [ncnt] = cur ;        cur = go [cur] [k] ;    }}void Build_fail ( )  {    std :: queue < int > q ;    for ( int i = 0 ; i < 26 ; ++ i )        if ( go [0] [i] )  q.push ( go [0] [i]) ;    while ( ! q.empty ( ) )  {        int cur = q.front ( ) ;  q.pop ( ) ;        for ( int i = 0 ; i ^ 26 ; ++ i )  {            if ( go [cur] [i] )  {                q.push ( go [cur] [i] ) ;                fail [go [cur] [i]] = go [fail [cur]] [i] ;            }  else  go [cur] [i] = go [fail [cur]] [i] ;        }    }}int in [N], out [N], idx ;void Dfs ( int u )  {    in [u] = ++ idx ;    for ( int i = head [u] ; i ; i = g [i].nxt )        Dfs ( g [i].to ) ;    out [u] = ++ idx ;}int c [N << 1], rt ;inline void Modify ( int pos, int delta )  {    while ( pos <= idx ) c [pos] += delta, pos += pos & -pos ;}inline int Query ( int pos )  {    for ( rt = 0 ; pos ; pos -= pos & -pos )    rt += c [pos] ;    return rt ;}int ans [N] ;int main ( )  {    scanf ( "%s", buf ) ;    Extend ( buf ) ;    Build_fail ( ) ;    for ( int i = 1 ; i <= ncnt ; ++ i ) add_edge ( fail [i], i ) ;    Dfs ( 0 ) ;    int n ;    scanf ( "%d", & n ) ;    for ( int i = 1 ; i <= n ; ++ i )  {        int x, y ;  // x in y ;        scanf ( "%d%d", & x, & y ) ;        qry [y].push_back ( ( Q ) { x, i } ) ;    }    int cur ( 0 ), id ( 0 ) ;    for ( char* pt = buf ; *pt ; ++ pt )  {        if ( *pt == 'P' )  {            int scnt = qry [++ id].size ( ) ;            for ( int i = 0 ; i < scnt ; ++ i )  {                int x = pos [qry [id] [i].x], qid = qry [id] [i].qid ;                ans [qid] = Query ( out [x] ) - Query ( in [x] - 1 ) ;            }            continue ;        }        if ( *pt == 'B' )  {            Modify ( in [cur], -1 ) ;            cur = fa [cur] ;            continue ;        }        cur = go [cur] [*pt - 'a'] ;        Modify ( in [cur], 1 ) ;    }    for ( int i = 1 ; i <= n ; ++ i )  {        printf ( "%d\n", ans [i] ) ;    }}