2017西安交大ACM小学期 文本查找[AC自动机]

文本查找

发布时间: 2017年7月5日 00:10   最后更新: 2017年7月5日 13:47   时间限制: 1500ms   内存限制: 128M

描述

给定m种两两不同的关键词，并给定一段文本，问这段文本中有几种关键词出现（一种关键词出现多次只算一次）。

输入

多组输入数据。
每组数据第一行一个正整数m，表示有m个关键词。
接下来m行每行一个关键词，关键词仅包含小写字母。
最后一行为文本，仅包含小写字母。
每组数据保证关键词总长度不超过106，文本不超过106。
总字符输入量不超过107。

输出

对于每组数据，输出一行一个整数，表示答案。

样例输入1 复制 3 a aa b aa 样例输出1 2 AC自动机的裸题，这里要说明的一点就是，一种关键词出现多次只能算一次，这样的话，我们就可以在一个关键词匹配完成后，在Trie树相关位置打上一个标记，防止下次重复计数。

代码:

#include <cstdio> #include <cstring> #include <queue> using namespace std; const int MAXN = 1e6+7;; #define LETTER 26 struct Trie{int num, fail,match;int next[LETTER]; }pool[MAXN]; Trie* const trie = pool + 1; int cnt; void init(){cnt = 0;memset(pool, 0, 2 * sizeof(Trie));trie[0].fail = -1; } inline int convert(char c){return c - 'a'; } void build() {queue<int> q; q.push(0);while (!q.empty()){int t = q.front(); q.pop();for (int i = 0; i < LETTER; i++){int &cur = trie[t].next[i];if (cur){q.push(cur);trie[cur].fail = trie[trie[t].fail].next[i];trie[cur].match = trie[cur].num ? cur :trie[trie[cur].fail].match;}else cur = trie[trie[t].fail].next[i];}} } int search(char *s) {int ret = 0, cur = 0;for (int i = 0; s[i]; i++){cur = trie[cur].next[convert(s[i])];for (int temp = trie[cur].match; temp;temp = trie[trie[temp].fail].match){ret += trie[temp].num;if(!trie[temp].num) break;trie[temp].num = 0;}}return ret; } void insert(char s[]){int cur = 0;for(int i = 0;s[i];i++){int &pos = trie[cur].next[convert(s[i])];if(!pos){pos = ++cnt;memset(&trie[cnt],0,sizeof(Trie));}cur = pos;}trie[cur].num ++; } char pat[MAXN]; char str[MAXN]; int main(){int m;while(~scanf("%d",&m)){init();while(m--){scanf(" %s",pat);insert(pat);}build();scanf(" %s",str);int ans = search(str);printf("%d\n",ans);} }

创作挑战赛新人创作奖励来咯，坚持创作打卡瓜分现金大奖

总结

以上是生活随笔为你收集整理的2017西安交大ACM小学期 文本查找[AC自动机]的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。