AC自动机模板

标签：算法   字符串   

AC自动机模板……

/*
 *	AC自动机模板
 *		使用方法：
 *			1、init() : 初始化函数
 *			2、insert(str) : 插入字符串函数
 *			3、build() : 构建ac自动机
 *			4、query(str) :　返回出现的字符串个数
 *
 *		使用需注意事项：
 *			1、注意输入的字符的范围，需对Next和其二维大小及相关参数进行更改
 *			2、注意Next、Fail和End数组的大小，防止超内存过数组越界
 *			3、根据实际情况对模板中“ buf[i] - 'a' ” 进行更改，否则可能会数组越界
 *		此模板默认相关设置：
 *			1、短字符串总长度不超过500000
 *			2、输入字符串的内容只由小写字母a~z构成
 *			3、query()函数只统计匹配的个数
 *				PS:上述都需根据需要自己更改！！！
 */
#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <queue>
using namespace std;

int Next[500010][26], Fail[500010], End[500010];
int root, L;
int newnode() {
	for (int i = 0; i < 26; i++)
		Next[L][i] = -1;
	End[L++] = 0;
	return L - 1;
}
void init() {
	L = 0;
	root = newnode();
}
void insert(char buf[]) {
	int len = strlen(buf);
	int now = root;
	for (int i = 0; i < len; i++) {
		if (Next[now][buf[i] - 'a'] == -1)
			Next[now][buf[i] - 'a'] = newnode();
		now = Next[now][buf[i] - 'a'];
	}
	End[now]++;
}
void build() {
	queue<int>Q;
	Fail[root] = root;
	for (int i = 0; i < 26; i++) {
		if (Next[root][i] == -1)
			Next[root][i] = root;
		else {
			Fail[Next[root][i]] = root;
			Q.push(Next[root][i]);
		}
	}
	while ( !Q.empty() ) {
		int now = Q.front();
		Q.pop();
		for (int i = 0; i < 26; i++) {
			if (Next[now][i] == -1)
				Next[now][i] = Next[Fail[now]][i];
			else {
				Fail[Next[now][i]] = Next[Fail[now]][i];
				Q.push(Next[now][i]);
			}
		}
	}
}
int query(char buf[]) {
	int len = strlen(buf);
	int now = root;
	int res = 0;
	for (int i = 0; i < len; i++) {
		now = Next[now][buf[i] - 'a'];
		int temp = now;
		while ( temp != root ) {
			res += End[temp];
			End[temp] = 0;
			temp = Fail[temp];
		}
	}
	return res;
}


int main() {
	freopen("in.in", "r", stdin);
	freopen("out.out", "w", stdout);
	return 0;
}

AC自动机模板

标签：算法   字符串   

原文地址：http://blog.csdn.net/frosero/article/details/47430661