HDU 3065 AC自动机模板题

最后一道模板题。

下面把AC自动机小结一下。请看新文章.....

这题如果只开26的next的数组的话。

在查询大串时，要注意的if( s[i] )当不符合条件时，不能浅显的continue,而要把p=root;

不然..... 那些在A-Z之外的字符会直接略去，相当于大串中没有A-Z之外的字符，结果可想而知！！

#include<iostream>
#include<string>
#include<cstdio>
#define MAX 26
using namespace std;

char str[2222222];
char str0[1111][55];
int cnt[1111],queue[555555];
int root,tot;

struct node
{
int fail,id;
int next[MAX];
void init()
{
memset( next,0,sizeof(next) );
fail=-1,id=0;
}
}Tire[555555];

void init()
{
root=tot=0;
Tire[root].init();
memset( cnt,0,sizeof(cnt) );
}

void insert( char *s,int id )
{
int i=0,k;
int p=root;
while( s[i] )
{
k=s[i++]-'A';
if( !Tire[p].next[k] )
{
Tire[++tot].init();
Tire[p].next[k]=tot;
}
p=Tire[p].next[k];
}
Tire[p].id=id;
}

void build_ac_automation()
{
int head,tail;
head=tail=0;
queue[tail++]=root;
while( head<tail )
{
int cur=queue[head++];
for( int i=0;i<MAX;i++ )
{
if( Tire[cur].next[i] )
{
int son=Tire[cur].next[i];
int p=Tire[cur].fail;
if( cur==root )
Tire[son].fail=root;
else
Tire[son].fail=Tire[p].next[i];
queue[tail++]=son;
}
else
{
int p=Tire[cur].fail;
if( cur==root )
Tire[cur].next[i]=root;
else
Tire[cur].next[i]=Tire[p].next[i];
}
}
}
}

void query( char *s )
{
int i=0,k;
int p=root;
while( s[i] )
{
k=s[i++]-'A';
if( k<0||k>25 )
{
p=root;
continue;
}
while( !Tire[p].next[k]&&p!=root )
p=Tire[p].fail;
p=Tire[p].next[k];
if( p==-1 ) p=0;
int temp=p;
while( temp!=root )
{
cnt[Tire[temp].id]++;
temp=Tire[temp].fail;
}
}
}

int main()
{
int N;
while( scanf( "%d\n",&N )!=EOF )
{
init();
for( int i=1;i<=N;i++ )
{
scanf( "%s",&str0[i] );
insert(str0[i],i);
}
build_ac_automation();
scanf( "%s",&str );
query(str);
for( int i=1;i<=N;i++ ){
if( cnt[i] )
printf( "%s: %d\n",str0[i],cnt[i] );
}
}
return 0;
}