HDU 3065 病毒侵袭持续中
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Problem Description - 题目描述
小t非常感谢大家帮忙解决了他的上一个问题。然而病毒侵袭持续中。在小t的不懈努力下,他发现了网路中的“万恶之源”。这是一个庞大的病毒网站,他有着好多好多的病毒,但是这个网站包含的病毒很奇怪,这些病毒的特征码很短,而且只包含“英文大写字符”。当然小t好想好想为民除害,但是小t从来不打没有准备的战争。知己知彼,百战不殆,小t首先要做的是知道这个病毒网站特征:包含多少不同的病毒,每种病毒出现了多少次。大家能再帮帮他吗?
Input - 输入 |
Output - 输出 |
第一行,一个整数N(1<=N<=500),表示病毒特征码的个数。 |
按以下格式每行一个,输出每个病毒出现次数。未出现的病毒不需要输出。 |
Sample Input - 输入样例 |
Sample Output - 输出样例 |
3 |
AA: 2 |
Hint - 提示
题目描述中没有被提及的所有情况都应该进行考虑。比如两个病毒特征码可能有相互包含或者有重叠的特征码段。
计数策略也可一定程度上从Sample中推测。
【题解】
每次匹配的时候不清除即可,一般的AC自动机,坑点是多组输入,以及眼神不好……
【代码 C++】
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
int tr[][], iTR, fail[], sWord[], cnt[];
char word[][], text[];
int n;
void build(){
memset(tr, , sizeof(tr)); memset(fail, , sizeof(fail));
memset(cnt, , sizeof(cnt)); memset(sWord, , sizeof(sWord));
iTR = ;
int i, j, k, w; for (i = ; i <= n; ++i){
gets(word[i]);
for (j = k = ; word[i][j]; ++j){
w = word[i][j] - 'A';
if (!tr[k][w]) tr[k][w] = ++iTR;
k = tr[k][w];
}
cnt[k] = i;
}
}
void setFail(){
std::queue<int> q;
int i, j, now;
for (i = ; i < ; ++i) if (tr[][i]) q.push(tr[][i]);
while (!q.empty()){
now = q.front(); q.pop();
for (i = ; i < ; ++i){
if (j = tr[now][i]) q.push(j), fail[j] = tr[fail[now]][i];
else tr[now][i] = tr[fail[now]][i];
}
}
}
void fid(){
gets(text);
int i, j, temp;
for (i = j = ; text[i]; ++i){
if (text[i] < 'A' || 'Z' < text[i]){ j = ; continue; }
j = tr[j][text[i] - 'A'];
for (temp = j; temp; temp = fail[temp]) ++sWord[cnt[temp]];
}
}
void opt(){
for (int i = ; i <= n; ++i){
if (sWord[i]) printf("%s: %d\n", word[i], sWord[i]);
}
}
int main(){
while (~scanf("%d", &n)){
getchar();
build(); setFail(); fid();
opt();
}
return ;
}