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] ) ;
    }
}
    原文作者:Trie树
    原文地址: https://blog.csdn.net/simpsonk/article/details/76277886
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞