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

AC自动机是建立在KMP算法和Trie树的基础上

一、主要步骤

1. 将所有的模式串构建成一个Trie树
2. 对Trie树上的所有节点求失配指针（即从根节点出发，一个串是另一个串的后缀，画图比较好理解）
3. 利用失配指针对主串进行匹配

代码，附注释，应该会好理解一些：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
using namespace std;
const int maxn=2e6+10;
int trie[maxn][30];
int flag[maxn],fail[maxn],cnt;
void insert(char *s){//构建模式串的Trie树
  int root=0,len=strlen(s);
  for (int i = 0;i < len;i++){
    int next=s[i]-'a';
    if (!trie[root][next]) trie[root][next]=++cnt;
    root=trie[root][next];
  }
  flag[root]++;
}
void getfail(){//求Fail失配数组
  queue<int> q;
  for (int i = 0;i < 26;i++){//遍历第一层的２６个字母
    if (trie[0][i]){//如果有这个字母
      fail[trie[0][i]]=0;//fail指向０
      q.push(trie[0][i]);
    }
  }
  while (!q.empty()){
    int now=q.front();
    q.pop();
    for (int i = 0;i < 26;i++){
      if (trie[now][i]){//如果这一层有这个叶子节点
    fail[trie[now][i]]=trie[fail[now]][i];//这个叶子节点i的失配指针指向它父亲节点失配指针指向的节点的i儿子，即保证是同一个字母
    q.push(trie[now][i]);
      }
      else trie[now][i]=trie[fail[now]][i];//如果不存在这个i叶子节点，指向当前节点失配指针指向节点的i叶子节点，即保证是同一个字母
    }
  }
}
int query(char *s){//查询一共出现了多少模式串
  int now=0,ans=0,len=strlen(s);
  for (int i = 0;i < len;i++){//遍历文本串
    now=trie[now][s[i]-'a'];
    for (int j = now;j&&flag[j]!=-1;j=fail[j]){
      //一直向下寻找，直到匹配失败（失配指针指向根或者当前节点已经找过）
      ans+=flag[j];
      flag[j]=-1;
    }
  }
  return ans;
}
int n;
char s[maxn];
int main(){
  scanf ("%d",&n);
  for (int i = 0;i < n;i++) {
    scanf ("%s",s);
    insert(s);
  }
  fail[0]=0;
  getfail();
  scanf ("%s",s);
  printf("%d
",query(s));
  return 0;
}