zoukankan      html  css  js  c++  java
  • 【模板】AC自动机(多少个模式串在文本串里出现过)

    #include<iostream>
    #include<cstdio>
    #include<cstdlib>
    #include<cstring>
    #include<cmath>
    #include<queue>
    #include<algorithm>
    using namespace std;
    struct Tree//字典树
    {
         int fail;//失配指针
         int vis[26];//子节点的位置
         int end;//标记有几个单词以这个节点结尾
    }AC[1000000];//Trie树
    int cnt=0;//Trie的指针
    
    inline void Build(string s)
    {
            int l=s.length();
            int now=0;//字典树的当前指针
            for(int i=0;i<l;++i)//构造Trie树
            {
                    if(AC[now].vis[s[i]-'a']==0)//Trie树没有这个子节点
                       AC[now].vis[s[i]-'a']=++cnt;//构造出来
                    now=AC[now].vis[s[i]-'a'];//向下构造
            }
            AC[now].end+=1;//标记单词结尾
    }
    void Get_fail()//构造fail指针
    {
            queue<int> Q;//队列
            for(int i=0;i<26;++i)//第二层的fail指针提前处理一下
            {
                   if(AC[0].vis[i]!=0)
                   {
                       AC[AC[0].vis[i]].fail=0;//指向根节点
                       Q.push(AC[0].vis[i]);//压入队列
                   }
            }
            while(!Q.empty())//BFS求fail指针
            {
                  int u=Q.front();
                  Q.pop();
                  for(int i=0;i<26;++i)//枚举所有子节点
                  {
                            if(AC[u].vis[i]!=0)//存在这个子节点
                          {
                                    AC[AC[u].vis[i]].fail=AC[AC[u].fail].vis[i];
                                        //子节点的fail指针指向当前节点的
                                      //fail指针所指向的节点的相同子节点
                                    Q.push(AC[u].vis[i]);//压入队列
                          }
                          else//不存在这个子节点
                          AC[u].vis[i]=AC[AC[u].fail].vis[i];
                          //当前节点的这个子节点指向当
                          //前节点fail指针的这个子节点
                  }
            }
    }
    int AC_Query(string s)//AC自动机匹配
    {
            int l=s.length();
            int now=0,ans=0;
            for(int i=0;i<l;++i)
            {
                    now=AC[now].vis[s[i]-'a'];//向下一层
                    for(int t=now;t&&AC[t].end!=-1;t=AC[t].fail)//循环求解
                    {
                             ans+=AC[t].end;
                             AC[t].end=-1;
                    }
            }
            return ans;
    }
    int main()
    {
         int n;
         char s[1000000];
         scanf("%d",&n);
         for(int i=1;i<=n;++i)
         {
                 scanf("%s",s);
                 Build(s);
         }
         AC[0].fail=0;//结束标志
         Get_fail();//求出失配指针
         scanf("%s",s);//文本串
         printf("%d
    ",AC_Query(s));
         return 0;
    }
  • 相关阅读:
    逻辑运算(二)
    Rust基础类型
    Rust简单demo
    逻辑基础(一)
    webpack配置typescript项目
    Go并发原理
    JS实现取任意类型的数组交集并集方法的思考
    JS 可逆加密的一种实现
    所有中文复姓
    将中文姓名转为拼音英文名的一种实现
  • 原文地址:https://www.cnblogs.com/Whiteying/p/10072984.html
Copyright © 2011-2022 走看看