首页 > 其他 > 详细

AC自动机 洛谷P3808 模板

时间:2019-08-18 23:18:29      阅读:110      评论:0      收藏:0      [点我收藏+]

题目链接:https://www.luogu.org/problem/P3808

题意:给n个模式串和一个文本串,求有多少个模式串在文本串中出现过

AC自动机的裸题,AC自动机最难理解的地方在于fail指针,这里借用了博客https://blog.csdn.net/creatorx/article/details/71100840#commentBox以及https://www.cnblogs.com/cjyyb/p/7196308.html的部分讲解

然后这个博客图很多,讲的很清楚:https://www.luogu.org/blog/hicc0305/solution-p3808

fail是失配指针,注意是失配
意味着,如果我此时匹配失败,那么,我们就要到达这个指针指向的位置继续尝试匹配
所以,我们可以将失配指针指向的的节点理解为:
当前节点所代表的串,最长的、能与后缀匹配的,在TrieTrie中出现过的前缀所代表的节点。
所以,fail指针类似于kmpnext数组,只不过由单串变为了多串而已。

举例:插入abcd, bce, abd, cd,构建字典树

技术分享图片

  首先我们让与根节点直接相连的节点的fail直接指向root,为了让你更好的理解fail指针,我们以节点x,y,z为例,我们让从图中我们可以看出x节点的fail指向了y节点,y节点的fail指向了z节点,为什么会这样指,因为x节点表示字符串abc,而字典树中含有最长,且以c结尾,且是abc的后缀的字符串bc(以y节点结尾的),同理,以y节点表示的字符串是bc,而以c结尾,且是bc的后缀的最长字符串是c(以z节点结尾的)。这就是fail指针指向的目标,那么我们得到了这个fail指针在匹配中有什么用呢,我们还是用上面的那个图来举例说明一下,假设文本串是abce,通过字典树我们可以看出,通过abc,所以我们可以匹配到x节点,但是到后面,我们发现d与e不匹配,这时我们就需要用到当前节点的fail了,因为x的fail指向的是y节点,所以我们直接跳到y节点,这是发现y节点后面有e,匹配上了,所以单词bce就在文本串abce中被检测出来了。当然这只是最简单的一种情况。

这里再解释一下代码中在求fail指针时为什么当前结点不存在就指向当前结点fail指针指向的结点的对应结点处,因为我们在查询有多少个子串的时候是加上所有达到的结点的数目(这个数目是指以这个节点结尾的单词的数目,且只加一次)。当我们遍历文本串时,遍历到了一个不存在的结点,用这种方法就可以直接转向前缀就是当前后缀的字符串了。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
const int inf=0x3f3f3f3f;
typedef long long ll;
#define meminf(a) memset(a,0x3f,sizeof(a))
#define mem0(a) memset(a,0,sizeof(a));
struct node{
    int fail;//失配指针fail
    int vis[26];//子节点的位置,也就是字典树的那26个字母
    int num;//记录有多少个单词以这个节点结尾 
}AC[maxn];
char s[maxn];
int cnt=0;//Trie的指针 

void insert(char *s){
    int len=strlen(s);
    int now=0;//字典树的当前指针
    for(int i=0;i<len;i++){
        //Trie树没有这个子节点 
        if(AC[now].vis[s[i]-a]==0) AC[now].vis[s[i]-a]=++cnt;
        now=AC[now].vis[s[i]-a];
    }
    AC[now].num++; 
}

void get_fail(){
    queue<int> que;
    for(int i=0;i<26;i++){//把第二层的fail指针都设为0 
        if(AC[0].vis[i]!=0)
        {
            AC[AC[0].vis[i]].fail=0;
            que.push(AC[0].vis[i]);
        }            
    }
    while(!que.empty())
    {
        int u=que.front();que.pop();
        for(int i=0;i<26;i++){
            if(AC[u].vis[i]!=0){
                //如果当前结点的子节点存在,就将子节点的fail指针指向当前结点fail指针指向的结点的对应子节点处 
                AC[AC[u].vis[i]].fail=AC[AC[u].fail].vis[i];
                que.push(AC[u].vis[i]);
            }
            else AC[u].vis[i]=AC[AC[u].fail].vis[i];
            //否则直接将这个不存在的子节点指向当前结点fail指针指向结点的对应子节点处 
        }
    }
}

int AC_query(char* s){
    int len=strlen(s);
    int ans=0,now=0;
    for(int i=0;i<len;i++){
        now=AC[now].vis[s[i]-a];
        for(int t=now;t!=0&&AC[t].num!=-1;t=AC[t].fail){
            //能够抵达的结点都将该节点的单词数加上,但记得标记加过一次后就不要再加了
            //不加是因为因为这道题并不是统计所有子串出现的总数,而是有多少子串出现了 
            ans+=AC[t].num;
            AC[t].num=-1;
        }
    }
    return ans;
}
int main(){
    int n;scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s);
        insert(s);
    }
    AC[0].fail=0;//结束标志     
    get_fail(); //求出失配指针 
    scanf("%s",s);
    printf("%d\n",AC_query(s)); 
    return 0;
} 

 

AC自动机 洛谷P3808 模板

原文:https://www.cnblogs.com/qingjiuling/p/11373907.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!