病毒侵袭持续中 HDU - 3065
Problem Description
小t非常感谢大家帮忙解决了他的上一个问题。然而病毒侵袭持续中。在小t的不懈努力下,他发现了网路中的“万恶之源”。这是一个庞大的病毒网站,他有着好多好多的病毒,但是这个网站包含的病毒很奇怪,这些病毒的特征码很短,而且只包含“英文大写字符”。当然小t好想好想为民除害,但是小t从来不打没有准备的战争。知己知彼,百战不殆,小t首先要做的是知道这个病毒网站特征:包含多少不同的病毒,每种病毒出现了多少次。大家能再帮帮他吗?
Input
第一行,一个整数N(1<=N<=1000),表示病毒特征码的个数。
接下来N行,每行表示一个病毒特征码,特征码字符串长度在1—50之间,并且只包含“英文大写字符”。任意两个病毒特征码,不会完全相同。
在这之后一行,表示“万恶之源”网站源码,源码字符串长度在2000000之内。字符串中字符都是ASCII码可见字符(不包括回车)。
Output
按以下格式每行一个,输出每个病毒出现次数。未出现的病毒不需要输出。
病毒特征码: 出现次数
冒号后有一个空格,按病毒特征码的输入顺序进行输出。
Sample Input
3
AA
BB
CC
ooxxCC%dAAAoen…END
Sample Output
AA: 2
CC: 1
题意:输入n个模板串病毒,再输入一个网站源代码,按顺序输出每个病毒在网站源代码中出现的次数,若出现0次,则不输出。
思路:统计每个模版在文本串中出现次数,当用文本匹配时,每发现一个单词节点 i,就将对应的 res[val[i]]++(这样即可连续的查询病毒),最后按顺序输出即可,此外由于字符为 ASCII 码,因此字典树的第二维要开到 128(根据样例连续AAA代表AA病毒出现两次)
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define E 1e-9
#define INF 0x3f3f3f3f
#define LL long long
const int MOD=10007;
const int N=500000+5;
using namespace std;
int res[N];//记录结果
struct AC_Automata{
int tire[N][128];//字典树
int val[N];//字符串结尾标记
int fail[N];//失配指针
int last[N];//last[i]=j表j节点表示的单词是i节点单词的后缀,且j节点是单词节点
int tot;//编号
void init(){//初始化0号点
tot=1;
val[0]=fail[0]=last[0]=0;
memset(tire[0],0,sizeof(tire[0]));
}
void insert(char *s,int v){//构造trie与val数组,v需非0,表示一个单词节点
int len=strlen(s);
int root=0;
for(int i=0;i<len;i++){
int id=s[i];
if(tire[root][id]==0){
tire[root][id]=tot;
memset(tire[tot],0,sizeof(tire[tot]));
val[tot++]=0;
}
root=tire[root][id];
}
val[root]=v;
}
void build(){//构造fail与last
queue<int> q;
last[0]=fail[0]=0;
for(int i=0;i<128;i++){
int root=tire[0][i];
if(root!=0){
fail[root]=0;
last[root]=0;
q.push(root);
}
}
while(!q.empty()){//bfs求fail
int k=q.front();
q.pop();
for(int i=0;i<128;i++){
int u=tire[k][i];
if(u==0)
continue;
q.push(u);
int v=fail[k];
while(v && tire[v][i]==0)
v=fail[v];
fail[u]=tire[v][i];
last[u]=val[fail[u]]?fail[u]:last[fail[u]];
}
}
}
void print(int i){//递归打印与结点i后缀相同的前缀节点编号
if(val[i]){
res[val[i]]++;
print(last[i]);
}
}
void query(char *s){//匹配
int len=strlen(s);
int j=0;
for(int i=0;i<len;i++){
int id=s[i];
while(j && tire[j][id]==0)
j=fail[j];
j=tire[j][id];
if(val[j])
print(j);
else if(last[j])
print(last[j]);
}
}
}ac;
char P[1000][1000];
char T[2000000+10];
int main(){
int n;
while(scanf("%d",&n)!=EOF&&n){
memset(res,0,sizeof(res));
ac.init();
for(int i=1;i<=n;i++){
scanf("%s",P[i]);
ac.insert(P[i],i);
}
ac.build();
scanf("%s",T);
ac.query(T);
for(int i=1;i<=n;i++)
if(res[i])
printf("%s: %d\n",P[i],res[i]);
}
return 0;
}