BZOJ-1030: [JSOI2007]文本生成器(AC自动机+DP)

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1030

AC自动机上的DP,f[i][j]表示文本匹配到i,在AC自动机上匹配到j的且未匹配完任何单词的情况,那么答案就是总数减去所有不合法数啦~(注意负数)

代码(之前数据范围开小莫名T了N次):

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
  
using namespace std ;
  
#define MAXN 110
#define MAXV 10010
#define MAXM 110
#define mod 10007
  
struct node {
    int child[ 26 ] , fail ;
    bool flag ;
    node(  ) {
        memset( child , 0 , sizeof( child ) ) ;
        fail = 0 ;
        flag = false ;
    }
} T[ MAXV ] ;
  
int V = 0 , n , m ;
char s[ MAXN ] ;
  
struct edge {
    edge *next ;
    int t ;
} *head[ MAXV ] ;
  
void AddEdge( int s , int t ) {
    edge *p = new( edge ) ;
    p -> t = t , p -> next = head[ s ] ;
    head[ s ] = p ;
}
  
void dfs( int v , bool flag ) {
    T[ v ].flag = flag ;
    for ( edge *p = head[ v ] ; p ; p = p -> next ) {
        dfs( p -> t , flag | T[ p -> t ].flag ) ;
    }
}
  
queue < int > q ;
  
void build_fail(  ) {
    while ( ! q.empty(  ) ) q.pop(  ) ;
    for ( int i = 0 ; i < 26 ; ++ i ) if ( T[ 0 ].child[ i ] ) {
        T[ T[ 0 ].child[ i ] ].fail = 0 ;
        q.push( T[ 0 ].child[ i ] ) ;
    }
    while ( ! q.empty(  ) ) {
        int v = q.front(  ) ; q.pop(  ) ;
        for ( int i = 0 ; i < 26 ; ++ i ) if ( T[ v ].child[ i ] ) {
            int j = T[ v ].fail , t = T[ v ].child[ i ] ;
            for ( ; j && ! T[ j ].child[ i ] ; j = T[ j ].fail ) ;
            if ( T[ j ].child[ i ] ) j = T[ j ].child[ i ] ;
            T[ t ].fail = j ;
            q.push( T[ v ].child[ i ] ) ;
        }
    }
    memset( head , 0 , sizeof( head ) ) ;
    for ( int i = 0 ; i ++ < V ; ) AddEdge( T[ i ].fail , i ) ;
    dfs( 0 , false ) ;
}
  
int f[ MAXN ][ MAXV ] ;
  
void Dp(  ) {
    memset( f , 0 , sizeof( f ) ) ;
    f[ 0 ][ 0 ] = 1 ; 
    for ( int i = 0 ; i < m ; ++ i ) {
        for ( int j = 0 ; j <= V ; ++ j ) {
            if ( f[ i ][ j ] ) {
                for ( int k = 0 ; k < 26 ; ++ k ) {
                    int t = j ;
                    for ( ; t && ! T[ t ].child[ k ] ; t = T[ t ].fail ) ;
                    if ( T[ t ].child[ k ] ) t = T[ t ].child[ k ] ;
                    if ( ! T[ t ].flag ) {
                        ( f[ i + 1 ][ t ] += f[ i ][ j ] ) %= mod ;
                    }
                }
            }
        }
    }
}
  
void Solve(  ) {
    int ans = 0 , ret = 1 ;
    for ( int i = 0 ; i ++ < m ; ) ( ret *= 26 ) %= mod ;
    for ( int i = 0 ; i <= V ; ++ i ) ( ans += f[ m ][ i ] ) %= mod ;
    ans = ret - ans ;
    while ( ans < 0 ) ans += mod ;
    printf( "%d\n" , ans ) ;
}
  
int main(  ) {
    scanf( "%d%d" , &n , &m ) ;
    for ( int i = 0 ; i ++ < n ; ) {
        scanf( "%s" , s + 1 ) ;
        int t = 0 ;
        for ( int j = 0 ; j ++ < strlen( s + 1 ) ; ) {
            if ( ! T[ t ].child[ s[ j ] - 'A' ] ) {
                T[ t ].child[ s[ j ] - 'A' ] = ++ V ;
            }
            t = T[ t ].child[ s[ j ] - 'A' ] ;
        }
        T[ t ].flag = true ;
    }
    build_fail(  ) ;
    Dp(  ) ;
    Solve(  ) ;
    return 0 ;
}

    原文作者:AmadeusChan
    原文地址: https://www.jianshu.com/p/a98a9dfa0dee
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞