bzoj 1212: [HNOI2004]L语言

AC 自动机模板

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2000000;
const int N = 500;
int ch[N][26],tot,f[N], // fail 树上的父节点
len[N],                 
val[N],                 // 是否为终点
last[N];                // 上一个终点
void init(int n){
    tot = 0;
    memset( ch[0],0,sizeof( ch[0] ) );
}
void Insert( char* str ){
    int p = 0;int n = strlen( str );
    for( int i = 0;i < n;i++ ){
        int c = str[i]-'a';
        if( !ch[p][c] )  {                   // 新建节点时记得初始化
            ch[p][c] = ++tot;val[tot]= 0;last[tot] = 0;f[tot] = 0;
            memset( ch[tot],0,sizeof( ch[tot] ) );
        }
        p = ch[p][c];
    }
    len[p] = n;val[p] = 1;
}
queue<int> que;
void getfail(){
    for( int i = 0;i < 26;i++ ){
        if( ch[0][i] ) que.push( ch[0][i] );
    }
    while(que.size()){
        int x = que.front();
        que.pop();
        for( int c = 0;c < 26;c++ ){
            if( !ch[x][c] ){
                ch[x][c] = ch[ f[x] ][c];
                continue;
            }
            f[ ch[x][c] ] = ch[ f[x] ][c];
            last[ch[x][c]] = val[ f[ ch[x][c] ] ] ? f[ch[x][c]] : last[ f[ ch[x][c] ] ];
            que.push(ch[x][c]);
        }
    }
}
char str[maxn];
int dp[maxn],ans;
void find( char* str ){
    int n = strlen(str);
    int p = 0;
    for( int i = 0;i < n;i++ ){
        int c = str[i]-'a';
        p = ch[p][c];
        for( int cur = p;cur;cur = last[cur] )if( len[cur] ){
            dp[i+1] |= dp[ i+1-len[cur] ];
            if( dp[i+1] ){ ans=i+1;break; }
        }
    }
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for( int i = 1;i <= n;i++ ){
        scanf("%s",str);
        Insert(str);
    }
    getfail();
    for( int i = 0;i < m;i++ ){
        scanf("%s",str+1);
        int n = strlen(str+1);
        for( int i = 0;i <= n;i++ ) dp[i] = 0;dp[0] = 1;
        ans = 0;
        find( str+1 );
        printf("%d\n",ans);
    }
    return 0;
}