HDU-1277 全文检索 AC自动机

　　题目给定一个非常巨大的串，然后给定许多模式串去匹配，如果是用KMP的话那么母串会被扫描很多次，这样便会超时。

　　AC自动机，一种对于多串匹配的优化，其原理很想KMP，只不过，在KMP算法中，next值只在于与自身的前缀进行匹配，而这里则将next的范围扩展到其他串的前缀，很NB的想法啊。哦，对了，其与KMP的区别还有一些地方，每次匹配一个字符时都搜索到根节点，看在这途中是否存在一个完整的单词。时间有限，不详写了。

　　代码如下：

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <queue>
using namespace std;

char text[60050], str[100];

int rec[10000];

struct Node
{
int type;
Node *fail;
Node *child[10];
};

void getstr( char *str )
{
char c;
while( c= getchar(), c< '0'|| c> '9' )  ;
*( str++ )= c;
while( c= getchar(), c>= '0'&& c<= '9' )
{
*( str++ )= c;
}
*( str )= '\0';
}

Node *init()
{
Node *r= new Node;
r->type= 0;
r->fail= NULL;
memset( r->child, 0, sizeof( r->child ) );
return r;
}

void insert( Node *r, char *in, int type )
{
if( *in!= '\0' )
{
if( r->child[*in- '0']== NULL )
{
r->child[*in- '0']= init();
}
insert( r->child[*in- '0'], in+ 1, type );
}
else
{
r->type= type;
}
}

void creatfail( Node *r )
{
queue< Node * >q;
q.push( r );
while( !q.empty() )
{
Node *pos= q.front();
q.pop();
for( int i= 0; i< 10; ++i )
{
Node *obj= pos->fail;
if( pos->child[i] )
{
while( obj!= NULL )
{
if( obj->child[i]!= NULL )
{
pos->child[i]->fail= obj->child[i];
break;
}
obj=obj->fail;
}
if( obj== NULL )
{
pos->child[i]->fail= r;
}
q.push( pos->child[i] );
}
}
}
}

void acauto( Node *r, char *text, int rec[], int &cnt )
{
int len= strlen( text );
Node *obj= r;
for( int i= 1; i< len; ++i )
{
int type= text[i]- '0';
while( obj!= NULL )
{
if( obj->child[type]!= NULL )
{
obj= obj->child[type];
break;
}
obj= obj->fail;
}
if( obj== NULL )
{
obj= r;
}
else
{
Node *temp= obj;
while( temp!= r )
{
if( temp->type!= 0 )
{
rec[cnt++]= temp->type;
temp->type= 0;
}
temp= temp->fail;
}
}
}
}

int main()
{
int M, N;
while( scanf( "%d %d", &M, &N )!= EOF )
{
Node *r= init();
int cnt= 0;
memset( text, 0, sizeof( text ) );
memset( rec, 0, sizeof( rec ) );
for( int i= 1; i<= M; ++i )
{
getstr( str );
strcat( text, str );
}
for( int i= 1; i<= N; ++i )
{
int c;
scanf( " [Key No. %d]", &c );
getstr( str );
insert( r, str, i );
}
creatfail( r );
acauto( r, text, rec, cnt );
if( cnt== 0 )
{
puts( "No key can be found !" );
}
else
{
printf( "Found key:" );
for( int i= 0; i< cnt; ++i )
{
printf( " [Key No. %d]", rec[i] );
}
puts( "" );
}
}

}