AC自动机

最新推荐文章于 2024-12-03 15:15:25 发布

原创最新推荐文章于 2024-12-03 15:15:25 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tree #insert #query #build #search #struct

ACM回忆专栏收录该内容

47 篇文章

订阅专栏

建立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);
	}
}