码迷,mamicode.com
首页 > 其他好文 > 详细

HDOJ-2222(AC自动机+求有多少个模板串出现在文本串中)

时间:2019-08-10 11:35:42      阅读:72      评论:0      收藏:0      [点我收藏+]

标签:algorithm   ring   状态   模板   return   复杂度   set   tps   fail   

Keywords Search

HDOJ-2222

  • 本文是AC自动机的模板题,主要是利用自动机求有多少个模板出现在文本串中
  • 由于有多组输入,所以每组开始的时候需要正确的初始化,为了不出错
  • 由于题目的要求是有多少字符串出现过,而不是出现过多少次,所以出现过的模板串就不能再计数了,所欲需要置-1.
  • 不要忘记build函数应该在insert函数之后调用,也不要忘记调用。
//AC自动机,复杂度为O(|t|+m),t表示文本串的长度,m表示模板串的个数
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
const int N=1E6+6;
int n;
int tree[N][26];//trie树上的结点,tree[i][j]表示i结点后面加一条j的边所对应的的结点
int total;//总结点
int num[N];//num[i]表示结点i上对应的模板串的个数
int fail[N];//失配指针,fail[i]指向所有模板串的前缀中匹配当前状态的最长后缀,指向的是最长后缀(和当前状态的后缀是匹配的,即相同,不过要最长)
queue<int> q;
int idx(char c){//用来求字符c对应的编号(0-25)
    return c-'a';
}
void insert(string s){//类似于后缀树的插入一个模板串
    int u=0;
    for(int i=0;i<s.length();i++){
        if(!tree[u][idx(s[i])])
            tree[u][idx(s[i])]=++total;
        u=tree[u][idx(s[i])];
    }
    num[u]++;
}
void build(){//建AC自动机以及fail数组
    for(int i=0;i<26;i++){
        if(tree[0][i])
            q.push(tree[0][i]);
    }
    while(!q.empty()){
        int u=q.front();
        q.pop();
        for(int i=0;i<26;i++){
            if(tree[u][i]){//如果结点u连的边为i对应的结点存在,则将这个存在的结点的fail指针指向父节点失配指针指向的结点的连的边为i所对应的的结点
                fail[tree[u][i]]=tree[fail[u]][i];
                q.push(tree[u][i]);
            }else{//类似于状态压缩,不至于每次fail指针跳转很多次,只需每次跳转一次,相当于构建了图
                tree[u][i]=tree[fail[u]][i];
            }
        }
    }
}
int query(string t){//s为要查找的文本串
    int u=0;
    int res=0;//记录答案,所有的模板串共出现了多少次
    for(int i=0;i<t.length();i++){
        u=tree[u][idx(t[i])];
        for(int j=u;j>0&&num[j]!=-1;){
            res+=num[j];
            num[j]=-1;
            j=fail[j];
        }
    }
    return res;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t;
    cin>>t;
    while(t--){
        cin>>n;
        while(!q.empty()){//清空队列
            q.pop();
        }
        memset(fail,0,sizeof(fail));
        memset(num,0,sizeof(num));
        memset(tree,0,sizeof(tree));
        string s;
        for(int i=0;i<n;i++){
            cin>>s;
            insert(s);
        }
        build();
        cin>>s;//模板串
        int ans=query(s);
        cout<<ans<<endl;
    }
    return 0;
}

HDOJ-2222(AC自动机+求有多少个模板串出现在文本串中)

标签:algorithm   ring   状态   模板   return   复杂度   set   tps   fail   

原文地址:https://www.cnblogs.com/GarrettWale/p/11330771.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!