病毒侵袭持续中 HDU - 3065
题意:给你n个模式串,1个文本串。统计各个模式串在文本串中出现的次数。
解题思路:
将模式串添加进入字典树时,纪录每个模式串在字典树上终点的位置。对文本串进行匹配时,每匹配到一个模式串,那个模式串出现的数量增加1。
坑点:
- 多组输入。没写多组输入的话返回的是wa
- 初始化模式串在字典树上终点的位置。虽然说后面的将模式串添加进字典树时,会自动覆盖。但是有可能部分没有被覆盖的值会在进行文本串匹配时出现bug。如果没有初始化这个数组会返回MLE而不是WA。因为答案可能会溢出,于是变成了负数,所以返回的是MLE。
AC代码:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
using namespace std;
//#pragma GCC optimize(3,"Ofast","inline")
#define LL long long
#define MT(a,b) memset(a,b,sizeof(a))
const int mod=1000007;
const int maxn=2e6+10;
const int ONF=-0x3f3f3f3f;
const int INF=0x3f3f3f3f;
char t[1005][55],s[maxn];
int trie[50005][26],id[50005],fail[50005],vis[1005],tot;
void add_trie(char *str,int n){
int len=strlen(str);
int pos=0;
for (int i=0;i<len;i++){
if (!trie[pos][str[i]-'A']) trie[pos][str[i]-'A']=++tot;
pos=trie[pos][str[i]-'A'];
}
id[pos]=n;
}
void get_fail(){
queue<int>q;
for (int i=0;i<26;i++){
if (trie[0][i]) {
q.push(trie[0][i]);
fail[trie[0][i]]=0;
}
}
while (!q.empty()){
int pos=q.front();
q.pop();
for (int i=0;i<26;i++){
if (trie[pos][i]){
fail[trie[pos][i]]=trie[fail[pos]][i];
q.push(trie[pos][i]);
} else trie[pos][i]=trie[fail[pos]][i];
}
}
}
void build_trie(int n){
MT(trie,0);
MT(vis,0);
MT(id,0);
tot=0;
for (int i=1;i<=n;i++){
scanf("%s",t[i]);
add_trie(t[i],i);
}
get_fail();
}
void solve(char *str){
int len=strlen(str);
for (int i=0,pos=0;i<len;i++){
int k=str[i]-'A';
if (k<0||k>25) {
pos=0;
continue;
}
pos=trie[pos][k];
for (int j=pos;j;j=fail[j]){
vis[id[j]]++;
}
}
}
int main(){
int n;
while (~scanf("%d",&n)){
build_trie(n);
scanf("%s",s);
solve(s);
for (int i=1;i<=n;i++){
if (!vis[i]) continue;
printf("%s",t[i]);
printf(": %d\n",vis[i]);
}
}
return 0;
}