建立AC自动机的一般步骤是:1、初始化根节点,根节点是所有字符串的前缀 2、利用模式串建立字典树(一般将主串叫匹配串,子串或去匹配的串叫模式串) 3、对字典树上的构建fail指针,fail指针指向当前串的最长后缀,这个后缀也是某个串的前缀,和KMP的next指针相似 4、利用构建好的ac自动机或者trie图(ac自动机的所有后继节点拓展之后就是trie图)进行操作,一般有查询、利用trie图建立矩阵、利用trie图进行状态DP等等。
自动机模板:
数据: 第一行1个整数代表T,第二行1个整数代表n个字符串,接下来n行每行一个字符串, 最后一个主串。查询有多少字符串能在主串中找到完全匹配
#include <stdio.h>
#include <string.h>
#include <iostream>
using namespace std;
#define MIN 11000
#define MAX 510000
struct node {
int cnt,flag;
node *next[26],*fail;
}tree[MAX],*root,*qu[MAX];
int n,ans,ptr,head,tail;
char str[MAX*2],dir[60];
node *CreateNode() {
for (int i = 0; i < 26; ++i)
tree[ptr].next[i] = NULL;
tree[ptr].fail = NULL;
tree[ptr].cnt = tree[ptr].flag = 0;
return &tree[ptr++];
}
void Initial() {
ans = ptr = 0;
head = tail = 0;
root = CreateNode();
}
void Insert(char *str) {
int i = 0,k;
node *p = root;
while (str[i]) {
k = str[i++] - 'a';
if (p->next[k] == NULL)
p->next[k] = CreateNode();
p = p->next[k];
}
p->cnt++,p->flag = 1;
}
void Build_AC() {
qu[head++] = root;
while (tail < head) {
node *p = qu[tail++];
for (int k = 0; k < 26; ++k)
if (p->next[k]) {
if (p == root) p->next[k]->fail = root;
else {
p->next[k]->fail = p->fail->next[k];
if (!p->next[k]->flag) p->next[k]->cnt = 1;
//象征性地表示以当前点的所有后缀是否有危险节点
}
qu[head++] = p->next[k];
}
else {
if (p == root) p->next[k] = root;
else p->next[k] = p->fail->next[k];
}
}
}
int Query(char *str) {
node *p = root;
int i = 0,k,cnt = 0;
while (str[i]) {
k = str[i++] - 'a';
p = p->next[k];
if (p->cnt) {
node *temp = p;
while (temp != root) {
if (temp->flag) cnt += temp->cnt;
temp->cnt = 0,temp = temp->fail;
}
}
}
return cnt;
}
int main()
{
int i,j,k,t;
scanf("%d",&t);
while (t--) {
scanf("%d",&n);
Initial();
for (i = 1; i <= n; ++i)
scanf("%s",dir),Insert(dir);
Build_AC();
scanf("%s",str);
ans = Query(str);
printf("%d\n",ans);
}
}