[BZOJ4327]-[JSOI2012]玄武密码-AC自动机

说在前面

并没有什么好说的…但是要保持格式!
看这个题比较顺眼于是就去切掉了…

---

题目

BZOJ4327传送门

题目大意

给出一个长串，称之为母串，再给出由很多短串组成的字典。
对于每个短串，需要处理出「该短串的前缀」在「母串」上的最大匹配长度（如果该短串被包含，那么这个长度就是短串长）
短串和长串的字符集均只有「 E 」「 S 」「 W 」「 N 」

输入输出格式

第一行有两个整数，N和M，分别表示母串的长度和短串的个数。 N≤1e7，M≤1e5
第二行是一个长度为N的字符串，表示母串。
之后M行，每行有一个字符串，表示短串，短串长小于等于100。

---

解法

很简单的AC自动机题=w=

求一个短串是否在一个长串中出现过，可以把短串的fail处理出来跑KMP。求最大匹配前缀长度的话，在KMP里随便加个变量记录一下就ok

然而数据规模吓死人，显然是不可能一个一个KMP的= =
那么显然需要用到AC自动机（废话，字典都给出来了），建立出fail指针之后，把母串拿到AC自动机上跑一遍。每到一个节点，就给这个节点打上vis标记。凡是AC自动机上被vis过的节点，就说明这个节点所代表的前缀被包含了。同时如果一个点被vis，那么它的fail也应该有vis标记。因此跑完之后，再自底向上更新vis标记。
最后把字典里每一个短串在trie上查一下最深的被vis过的点就ok了

---

下面是自带大常数的代码

/**************************************************************    Problem: 4327    User: Izumihanako    Language: C++    Result: Accepted    Time:3428 ms    Memory:188740 kb****************************************************************/ #include <queue>#include <cstdio>#include <cstring>#include <algorithm>using namespace std ; int N , M ;char ss[100005][105] , a[10000005] ;struct Node{    bool vis ;    char c ;    Node *ch[5] , *fail ;}*root ; void newNode( Node *&nd ){    nd = new Node() ;    for( int i = 0 ; i < 5 ; i ++ )        nd->ch[i] = NULL ;    nd->fail = NULL ; nd->vis = false ;} void ext( char *ts ,int len ){    for( int i = 0 ; i < len ; i ++ )        switch( ts[i] ){            case 'E' : ts[i] = 1 ; break ;            case 'S' : ts[i] = 2 ; break ;            case 'W' : ts[i] = 3 ; break ;            case 'N' : ts[i] = 4 ; break ;        }} void Insert( char *ts ){    int len = strlen( ts ) ;    Node *nd = root ;    for( int i = 0 ; i < len ; i ++ ){        int nxt = ts[i] ;        if( !nd->ch[nxt] ){            newNode( nd->ch[nxt] ) ;            //printf( "address :%d\n" , nd->ch[nxt] ) ;            nd->ch[nxt]->c = ts[i] ;        }        nd = nd->ch[nxt] ;    }} queue<Node*> que ;void getFail(){    que.push( root ) ;    while( !que.empty() ){        Node *u = que.front() ; que.pop() ;        for( int i = 1 ; i < 5 ; i ++ ){            if( !u->ch[i] ) continue ;            Node *p = u->fail , *v = u->ch[i] ;            while( p && !p->ch[i] ) p = p->fail ;            v->fail = ( p ? p->ch[i] : root ) ;               que.push( v ) ;        }    }    root->fail = root ;} void Run(){    Node *nd = root ; root->vis = true ;    for( int i = 0 ; i < N ; i ++ ){        int nxt = a[i] ;        //printf( "adress(%d-[%d]) nxt :%d\n" , nd , nd->c , nxt ) ;        while( nd != root && !nd->ch[nxt] ){            nd = nd->fail ;            nd->vis = true ;        }        if( nd->ch[nxt] ){            nd = nd->ch[nxt] ;            nd->vis = true ;        }    }} Node *sta[40000005] ;void bfs(){    int pt = 0 , topp = 0 ;    sta[++topp] = root ;    while( pt != topp ){        pt ++ ;        Node *u = sta[pt] ;        for( int i = 1 ; i < 5 ; i ++ )            if( u->ch[i] ) sta[++topp] = u->ch[i] ;    }    for( int i = topp ; i ; i -- ){        Node *u = sta[i] ;        if( u->vis ) u->fail->vis = true ;    }} int Query( char *ts ){    int len = strlen( ts ) , rt = 0 ;    Node *nd = root ;    for( int i = 0 ; i < len ; i ++ ){        int nxt = ts[i] ;        nd = nd->ch[nxt] ;        if( !nd->vis ) return rt ;        rt ++ ;    }    return rt ;} void solve(){    getFail() ;    Run() ;    bfs() ;    for( int i = 1 ; i <= M ; i ++ )        printf( "%d\n" , Query( ss[i] ) ) ;} int main(){    scanf( "%d%d" , &N , &M ) ;    scanf( "%s" , a ) ;    ext( a , N ) ;    newNode( root ) ; root->c = 0 ;    for( int i = 1 ; i <= M ; i ++ ){        scanf( "%s" , ss[i] ) ;        ext( ss[i] , strlen( ss[i] ) ) ;        Insert( ss[i] ) ;    }    solve() ;}