J - 病毒侵袭持续中
小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
思路:
AC自动机模板题,设一个计数的数组存下答案。
AC旅程 :
WR,TLE,MLE 我已处于绝望的边缘,但愿下一题不要推向绝望?。
WR了6发,TLE5发,终于过了。。。
WR 错在:1)代码中vis数组没清零。 2)先把所有大写字母存在另一个数组里,再查询,导致这样的一组样例过不了 (ooxxCC%dAAAoen....ABCCD) 结果应是AA:2,CC:2; 因为先把大写字母存进去会把4个A连在一起。
TLE错在:在查询中,我把不是大写字母的 rt=0,continue; (网上有的这样写过来,反正我的代码这样写就TLE)
代码:
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
const int maxn=1e6;
const int maxm=5*1e4+5;
char text[2*maxn+5],tmp[2*maxn+5];
char mo[1010][55];
int n;
int book[1010];
struct Aho
{
int ne[maxm][26];
int fail[maxm];
int vis[maxm];
int tot,rt;
queue<int>qu;
void init()
{
while(!qu.empty())
qu.pop();
mem(ne[0],0);
mem(fail,0);
mem(vis,0);
tot=1;
}
void Insert(char *str,int y)
{
rt=0;
int ls=strlen(str);
for(int i=0; i<ls; i++)
{
int x=str[i]-'A';
if(!ne[rt][x])
{
mem(ne[tot],0);
ne[rt][x]=tot++;
}
rt=ne[rt][x];
}
vis[rt]=y;
}
void Build()
{
fail[0]=-1;
qu.push(0);
while(!qu.empty())
{
int u=qu.front();
qu.pop();
for(int i=0; i<26; i++)
{
if(ne[u][i])
{
if(u==0)
fail[ne[u][i]]=0;
else
{
int v=fail[u];
while(v!=-1)
{
if(ne[v][i])
{
fail[ne[u][i]]=ne[v][i];
break;
}
v=fail[v];
}
if(v==-1)
fail[ne[u][i]]=0;
}
qu.push(ne[u][i]);
}
}
}
}
void Match(char *str)
{
rt=0;
int ls=strlen(str);
for(int i=0; i<ls; i++)
{
int x=str[i]-'A';
if(ne[rt][x])
rt=ne[rt][x];
else //找祖先
{
int f=fail[rt];
while(f!=-1 && ne[f][x]==0) //所指数没有值
f=fail[f];
if(f==-1) //始终没找到,回到根节点
rt=0;
else
rt=ne[f][x];
}
int lin=rt;
while(lin!=0)
{
if(vis[lin])
{
book[vis[lin]]++;
}
lin=fail[lin];
}
}
}
} aho;
int main()
{
while(~scanf("%d",&n))
{
mem(book,0);
aho.init();
for(int i=1; i<=n; i++)
{
scanf("%s",mo[i]);
aho.Insert(mo[i],i);
}
aho.Build();
scanf("%s",text);
int ls=strlen(text);
int t=0;
for(int i=0;i<=ls;i++)
{
if(text[i]>='A'&&text[i]<='Z')
{
tmp[t++]=text[i];
}
else
{
tmp[t]='\0';
aho.Match(tmp);
t=0;
}
}
for(int i=1; i<=n; i++)
{
if(book[i])
{
printf("%s: %d\n",mo[i],book[i]);
}
}
}
return 0;
}