建立AC自动机的一般步骤是:1、初始化根节点,根节点是所有字符串的前缀 2、利用模式串建立字典树(一般将主串叫匹配串,子串或去匹配的串叫模式串) 3、对字典树上的构建fail指针,fail指针指向当前串的最长后缀,这个后缀也是某个串的前缀,和KMP的next指针相似 4、利用构建好的ac自动机或者trie图(ac自动机的所有后继节点拓展之后就是trie图)进行操作,一般有查询、利用trie图建立矩阵、利用trie图进行状态DP等等。
自动机模板:
hdu 2222 Keywords Search
数据: 第一行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); } }