模板】AC自动机（简单版）

模板】AC自动机（简单版）

https://www.luogu.org/problemnew/show/P3808

这是一道简单的AC自动机模板题。

用于检测正确性以及算法常数。

为了防止卡OJ，在保证正确的基础上只有两组数据，请不要恶意提交。

管理员提示：本题数据内有重复的单词，且重复单词应该计算多次，请各位注意

题目描述

给定n个模式串和1个文本串，求有多少个模式串在文本串里出现过。

输入输出格式

输入格式：

第一行一个n，表示模式串个数；

下面n行每行一个模式串；

下面一行一个文本串。

输出格式：

一个数表示答案

输入输出样例

输入样例#1

2
a
aa
aa

输出样例#1

2

说明

subtask1[50pts]:∑length(模式串)<=10^6,length(文本串)<=10^6,n=1；

subtask2[50pts]:∑length(模式串)<=10^6,length(文本串)<=10^6；

参考博客：http://www.cnblogs.com/cjyyb/p/7196308.html

听说有一种需要用trie树做，trie图不能做的题目？？先mark下

关于失配指针的描述：从当前节点开始，沿着其父节点的失配指针不断向上跑，直到到达一个节点，它的儿子中有当前字母，然后把这两个一样的字母连起来。

首先要构建trie图，然后fail是在trie图上实现的，所以fail上的跳转是在模式串上的跳转，通过这样的跳转，可以快速找到相匹配的模式串

 #include<bits/stdc++.h>
 using namespace std;
 #define lson l,mid,rt<<1
 #define rson mid+1,r,rt<<1|1
 #define sqr(x) ((x)*(x))
 #define maxn 1000005
 typedef long long ll;
 typedef unsigned long long ull;
 const ull MOD=;
 /*#ifndef ONLINE_JUDGE
         freopen("1.txt","r",stdin);
 #endif */

 struct tree{
     int fail;
     int vis[];
     int num;
 }ac[];

 int cnt=;

 void build(string s){///建trie树
     int len=s.length();
     int now=;
     for(int i=;i<len;i++){
         if(ac[now].vis[s[i]-'a']==){
             ac[now].vis[s[i]-'a']=++cnt;
         }
         now=ac[now].vis[s[i]-'a'];
     }
     ac[now].num+=;
 }

 void get_fail(){///构建成trie图
     queue<int>Q;
     for(int i=;i<;i++){
         if(ac[].vis[i]){
             ac[ac[].vis[i]].fail=;
             Q.push(ac[].vis[i]);
         }
     }
     while(!Q.empty()){
         int u=Q.front();
         Q.pop();
         for(int i=;i<;i++){
             if(ac[u].vis[i]){
                 ac[ac[u].vis[i]].fail=ac[ac[u].fail].vis[i];
                 Q.push(ac[u].vis[i]);
             }
             else{
                 ac[u].vis[i]=ac[ac[u].fail].vis[i];///如果当前结点不存在，就指向父亲结点的fail指向的结点的子结点
             }
         }
     }
 }

 int ac_query(string s){
     int len=s.length();
     int now=,ans=;
     for(int i=;i<len;i++){
         now=ac[now].vis[s[i]-'a'];
         for(int t=now;t&&ac[t].num!=-;t=ac[t].fail){
             ans+=ac[t].num;
             ac[t].num=-;
         }
     }
     return ans;
 }

 int main(){
     #ifndef ONLINE_JUDGE
       //  freopen("1.txt","r",stdin);
     #endif
     //std::ios::sync_with_stdio(false);
     string s;
     int n;
     cin>>n;
     for(int i=;i<n;i++){
         cin>>s;
         build(s);
     }
     ac[].fail=;
     get_fail();
     cin>>s;
     cout<<ac_query(s)<<endl;
 }