近期一直在学习字符串之类的算法,感觉BF算法,尽管非常easy理解,可是easy超时,全部就想学习其它的一些字符串算法来提高一下,近期学习了一下AC自己主动机。尽管感觉有所收获,可是还是有些朦胧的感觉,在此总结一下,希望大家不吝赐教。
一、AC自己主动机的原理:
Aho-Corasick automaton。该算法在1975年产生于贝尔实验室,是著名的多模匹配算法之中的一个。
一个常见的样例就是给出N个单词,在给出一段包括m个字符的文章,让你找出有多少个单词在这文章中出现过,。要搞懂AC自己主动机。先的有字典树和KMP模式匹配算法的基础知识。
假设没有kmp或者字典树算法基础的能够看看:
<span style="font-size:18px;">//kmp http://blog.csdn.net/qq_16997551/article/details/51038525</span>
<span style="font-size:18px;">//字典树 http://blog.csdn.net/qq_16997551/article/details/51107243</span>
二、AC自己主动机算法的实现步骤(三步)
AC自己主动机的存储数据结构
const int MAXN = 10000000; struct node { int count; //是否为单词最后一个节点 node *next[26];//Trie每一个节点的26个子节点 node *fail; //失败指针 }; node *q[MAXN]; //队列。採用bfs 构造失败指针 char keyword[55];//输入单词 模式串 char str[1000010];// 须要查找的 主串 int head,tail;//队列 头尾指针
1、构造一棵Trie树
首先我们须要建立一棵Trie。
可是这棵Trie不是普通的Trie,而是带有一些特殊的性质。
首先会有3个重要的指针,分别为p, p->fail, temp。
1.指针p,指向当前匹配的字符。若p指向root,表示当前匹配的字符序列为空。
(root是Trie入口。没有实际含义)。
2.指针p->fail,p的失败指针,指向与字符p同样的结点,若没有。则指向root。
3.指针temp,測试指针(自己命名的。easy理解!~),在建立fail指针时有寻找与p字符匹配的结点的作用,在扫描时作用最大,也最不好理解。
对于Trie树中的一个节点,相应一个序列s[1...m]。此时,p指向字符s[m]。若在下一个字符处失配,即p->next[s[m+1]] == NULL,则由失配指针跳到还有一个节点(p->fail)处,该节点相应的序列为s[i...m]。若继续失配,则序列依次跳转直到序列为空或出现匹配。在此过程中。p的值一直在变化,可是p相应节点的字符没有发生变化。
在此过程中,我们观察可知,终于求得得序列s则为最长公共后缀。另外。因为这个序列是从root開始到某一节点,则说明这个序列有可能是某些序列的前缀。
再次讨论p指针转移的意义。假设p指针在某一字符s[m+1]处失配(即p->next[s[m+1]] == NULL),则说明没有单词s[1...m+1]存在。
此时。假设p的失配指针指向root,则说明当前序列的随意后缀不会是某个单词的前缀。
假设p的失配指针不指向root,则说明序列s[i...m]是某一单词的前缀,于是跳转到p的失配指针。以s[i...m]为前缀继续匹配s[m+1]。
对于已经得到的序列s[1...m],因为s[i...m]可能是某单词的后缀,s[1...j]可能是某单词的前缀,所以s[1...m]中可能会出现单词。此时,p指向已匹配的字符,不能动。
于是。令temp = p。然后依次測试s[1...m], s[i...m]是否是单词。
构造的Trie为:
实现代码:
<span style="font-size:18px;">void insert(char *word,node *root) { int index,len; node *p = root,*newnode; len = strlen(word); for(int i=0 ;i < len ; i++ ) { index=word[i]-'a'; if(!p->next[index])//该字符节点不存在。增加Trie树中 { // 初始化 newnode 并 增加 Trie 树 newnode=(struct node *)malloc(sizeof(struct node)); for(int j=0;j<26;j++) newnode->next[j]=0; newnode->count=0; newnode->fail=0; p->next[index]=newnode; } p=p->next[index];//指针移动至下一层 } p->count++; //单词结尾 节点 count + 1 做标记 }</span>
2、构造失败指针
构造失败指针的过程概括起来就一句话:设这个节点上的字母为x,沿着他父亲的失败指针走,直到走到一个节点,他的儿子中也有字母为x的节点。
然后把当前节点的失败指针指向那个字符也为x的儿子。
假设一直走到了root都没找到,那就把失败指针指向root。
有两个规则:
-
root的子节点的失败指针都指向root。
-
节点(字符为x)的失败指针指向:从X节点的父节点的fail节点回溯直到找到某节点的子节点也是字符x。没有找到就指向root。
例如以下图
实现代码:
<span style="font-size:18px;">void build_ac_automation(node *root) { head=0; tail=1; q[head]=root; node *temp,*p; while(head<tail)//bfs构造 Trie树的失败指针 { //算法相似 kmp ,这里相当于得到 next[]数组 //重点在于,匹配失败时,由fail指针回溯到正确的位置 temp=q[head++]; for(int i=0;i< 26 ;i ++) { if(temp->next[i])//推断实际存在的节点 { // root 下的第一层 节点 的 失败指针都 指向root if(temp==root) temp->next[i]->fail=root; else { //依次回溯 该节点的父节点的失败指针 //直到某节点的next[i]与该节点同样。则 //把该节点的失败指针指向该next[i]节点 //若回溯到 root 都没有找到,则该节点 //的失败指针 指向 root p=temp->fail;//temp 为节点的父指针 while(p) { if(p->next[i]) { temp->next[i]->fail=p->next[i]; break; } p=p->fail; } if(!p)temp->next[i]->fail=root; } //每处理一个点,就把它的全部儿子增加队列, //直到队列为空 q[tail++]=temp->next[i]; } } } }</span>
3、模式匹配过程
从root节点開始,每次依据读入的字符沿着自己主动机向下移动。
当读入的字符。在分支中不存在时,递归走失败路径。假设走失败路径走到了root节点, 则跳过该字符。处理下一个字符。 由于AC自己主动机是沿着输入文本的最长后缀移动的,所以在读取全然部输入文本后,最后递归走失败路径,直到到达根节点, 这样能够检測出全部的模式。
搜索的步骤:
-
从根节点開始一次搜索;
-
取得要查找关键词的第一个字符。并依据该字符选择相应的子树并转到该子树继续进行检索;
-
在相应的子树上,取得要查找关键词的第二个字符,并进一步选择相应的子树进行检索。
-
迭代过程……
-
在某个节点处。关键词的全部字符已被取出,则读取附在该节点上的信息,即完毕查找。
匹配模式串中出现的单词。当我们的模式串在Trie上进行匹配时,假设与当前节点的keyword不能继续匹配的时候。
就应该去当前节点的失败指针所指向的节点继续进行匹配。
匹配过程出现两种情况:
-
当前字符匹配,表示从当前节点沿着树边有一条路径能够到达目标字符, 此时仅仅需沿该路径走向下一个节点继续匹配就可以 。目标字符串指针移向下个字符继续匹配;
-
当前字符不匹配,则去当前节点失败指针所指向的字符继续匹配,匹配过程随着指针指向root结束。
反复这2个过程中的随意一个。直到模式串走到结尾为止。
实现代码:
<span style="font-size:18px;">int query(node *root)//相似于 kmp算法。 {//i为主串指针,p为匹配串指针 int i,cnt=0,index,len=strlen(str); node *p=root; for(i=0; i < len ;i ++) { index=str[i]-'a'; //由失败指针回溯寻找,推断str[i]是否存在于Trie树中 while( !p->next[index] && p != root) { p=p->fail; } p=p->next[index];//找到后 p 指向该节点 //指针回为空。则没有找到与之匹配的字符 if(!p) { p=root;//指针又一次回到根节点root,下次从root開始搜索Trie树 } node *temp=p;//匹配该节点后。沿其失败指针回溯,推断其它节点是否匹配 while(temp != root )//匹配 结束控制 { if(temp->count>=0)//推断 该节点是否被訪问 { //统计出现的单词个数cnt。因为节点不是单词结尾时count为0。 //故 cnt+=temp->count; 仅仅有 count >0时才真正统计了单词个数 cnt+=temp->count; temp->count=-1; //标记已訪问 } else break;//节点已訪问,退出循环 temp=temp->fail;//回溯失败指针继续寻找下一个满足条件的节点 } } return cnt; }</span>
三、AC自己主动机模板
<span style="font-size:18px;">#include<stdio.h> #include<string.h> #include<stdlib.h> #define kind 26 const int MAXN = 10000000; struct node { int count; //是否为单词最后一个节点 node *next[26];//Trie每一个节点的26个子节点 node *fail; //失败指针 }; node *q[MAXN]; //队列,採用bfs 构造失败指针 char keyword[55];//输入单词 模式串 char str[1000010];// 须要查找的 主串 int head,tail;//队列 头尾指针 node *root; void insert(char *word,node *root) { int index,len; node *p = root,*newnode; len = strlen(word); for(int i=0 ;i < len ; i++ ) { index=word[i]-'a'; if(!p->next[index])//该字符节点不存在,增加Trie树中 { // 初始化 newnode 并 增加 Trie 树 newnode=(struct node *)malloc(sizeof(struct node)); for(int j=0;j<26;j++) newnode->next[j]=0; newnode->count=0; newnode->fail=0; p->next[index]=newnode; } p=p->next[index];//指针移动至下一层 } p->count++; //单词结尾 节点 count + 1 做标记 } void build_ac_automation(node *root) { head=0; tail=1; q[head]=root; node *temp,*p; while(head<tail)//bfs构造 Trie树的失败指针 { //算法相似 kmp ,这里相当于得到 next[]数组 //重点在于,匹配失败时。由fail指针回溯到正确的位置 temp=q[head++]; for(int i=0;i< 26 ;i ++) { if(temp->next[i])//推断实际存在的节点 { // root 下的第一层 节点 的 失败指针都 指向root if(temp==root) temp->next[i]->fail=root; else { //依次回溯 该节点的父节点的失败指针 //直到某节点的next[i]与该节点同样。则 //把该节点的失败指针指向该next[i]节点 //若回溯到 root 都没有找到,则该节点 //的失败指针 指向 root p=temp->fail;//temp 为节点的父指针 while(p) { if(p->next[i]) { temp->next[i]->fail=p->next[i]; break; } p=p->fail; } if(!p)temp->next[i]->fail=root; } //每处理一个点,就把它的全部儿子增加队列。 //直到队列为空 q[tail++]=temp->next[i]; } } } } int query(node *root)//相似于 kmp算法。 {//i为主串指针,p为匹配串指针 int i,cnt=0,index,len=strlen(str); node *p=root; for(i=0; i < len ;i ++) { index=str[i]-'a'; //由失败指针回溯寻找,推断str[i]是否存在于Trie树中 while( !p->next[index] && p != root) { p=p->fail; } p=p->next[index];//找到后 p 指向该节点 //指针回为空。则没有找到与之匹配的字符 if(!p) { p=root;//指针又一次回到根节点root,下次从root開始搜索Trie树 } node *temp=p;//匹配该节点后,沿其失败指针回溯,推断其它节点是否匹配 while(temp != root )//匹配 结束控制 { if(temp->count>=0)//推断 该节点是否被訪问 { //统计出现的单词个数cnt,因为节点不是单词结尾时count为0。 //故 cnt+=temp->count; 仅仅有 count >0时才真正统计了单词个数 cnt+=temp->count; temp->count=-1; //标记已訪问 } else break;//节点已訪问,退出循环 temp=temp->fail;//回溯失败指针继续寻找下一个满足条件的节点 } } return cnt; } int main() { int i,t,n,ans; scanf("%d",&t); while(t--) { root=(struct node *)malloc(sizeof(struct node)); for(int j=0;j<26;j++) root->next[j]=0; root->fail=0; root->count=0; scanf("%d",&n); getchar(); for(i=0;i<n;i++) { gets(keyword); insert(keyword,root); } build_ac_automation(root); gets(str); ans=query(root); printf("%d ",ans); } return 0; }</span>