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

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/news/321665.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！