hdu2222 Keywords Search【AC自动机】

题意:

t组数据,每次给你n个模式串,和一个长串,请问这个长串里面包含几个模式串
n的范围为1e4,每个模式串的长度不超过50,长串的长度不超过1e6

题解:

AC自动机入门,也是我第一次写AC自动机
学习博客
模板用的是kuangbin的模板

要点:
fail数组初始化都为0,而我的trie的根节点也为0
套路总结:
先初始化空的自动机,插入每个模式串,建立字典树,对建立完的字典树跑一遍bfs建立fail指针

AC_code:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e5+5;
struct tree {
	int next[maxn][26], fail[maxn], end[maxn];
	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() { //使用bfs 建立fail指针
		queue<int> Q;
// cout<<root<<endl;
		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) {
// cout<<now<<" "<<fail[now]<<endl;
					next[now][i] = next[fail[now]][i];
				} else {
// cout<<now<<" "<<next[now][i]<<" "<<fail[next[now][i]]<<endl; 
					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;
	}
};
tree ac;
char buf[maxn*2];
int main() {
	int t, n;
	scanf("%d", &t);
	while(t--) {
		scanf("%d", &n);
		ac.init();
		for(int i = 0; i < n; i++) {
			scanf("%s", buf);
			ac.insert(buf);
		}
		ac.build();
		scanf("%s", buf);
		int ans = ac.query(buf);
		printf("%d\n", ans);
	}
}
全部评论

相关推荐

头像
11-07 01:12
重庆大学 Java
精致的小松鼠人狠话不多:签哪了哥
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务