HDU 2222 目标串中模式串数量(AC自动机)

题意:给定n(n≤50)个模式串(每个模式串长度不超过50)和1个目标串(长度不超过1000000),问目标串中包含多少个模式串。

思路:用模式串建立DFA,然后用目标串扫一遍即可。使用数组建立DFA,模板参考(http://www.cnblogs.com/kuangbin/p/3164106.html)。数组开500000一只TLE,减小到250000才能AC(但是链接中的数组开的就是500000呀。。。)

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <cstdlib>
using namespace std;
#define N 250010
char s[1000005];
queue<int> q;
int t[N][26],fail[N],num[N];
int T,n,m,top,res;
void init(){
    int i;
    top = 1;
    res = 0;
    memset(fail, 0, sizeof(fail));
    memset(t, -1, sizeof(t));
    memset(num, 0, sizeof(num));
    for(i = 0;i<26;i++)
        t[0][i] = 1;
}

void insert(char* s){//建立trie树,
    int i,r = 1;
    for(i = 0;s[i];i++){
        if(t[r][s[i]-'a']==-1)
            t[r][s[i]-'a'] = ++top;
        r = t[r][s[i]-'a'];
    }
    num[r] ++;
}
void buildDFA(){//构造DFA
    int i,now;
    q.push(1);
    while(!q.empty()){
        now = q.front();
        q.pop();
        for(i = 0;i<26;i++){
            if(t[now][i] == -1)
                t[now][i] = t[fail[now]][i];
            else{
                fail[t[now][i]] = t[fail[now]][i];
                q.push(t[now][i]);
            }
        }
    }
}
void search(char* s){
    int i,j,r = 1;
    for(i = 0;s[i];i++){
        j = r = t[r][s[i]-'a'];
        while(j){
            res += num[j];
            num[j] = 0;
            j = fail[j];
        }
    }
}
int main(){
    scanf("%d",&T);
    while(T--){
        int i;
        init();
        scanf("%d",&n);
        for(i = 0;i<n;i++){
            scanf("%s",s);
            insert(s);
        }
        buildDFA();
        scanf("%s",s);
        search(s);
        printf("%d\n",res);
    }
    return 0;
}


你可能感兴趣的:(HDU 2222 目标串中模式串数量(AC自动机))