hdu 3065 病毒侵袭持续中 (AC)

http://acm.hdu.edu.cn/showproblem.php?pid=3065 

AC自动机，主要注意的就是两个特征码的重叠情况。

code:

#include<iostream>
#include<cstring>
#include<cstdio>
using  namespace std ;
const  int kind =  30;
struct node{
    node *fail ;
    node *next[kind] ;
     int count ;
     int id ;
    node(){
       fail = NULL ;
       count =  0 ;
       id = - 1 ;
       memset(next, NULL,  sizeof(next)) ;
   }
}*q[ 500001] ;
char keyword[ 1001][ 51] ;
char str[ 2000001] ;
int head, tail, ID ;
int vis[ 1001] ;

void insert( char *str, node *root){
    node *p = root ;
     int i =  0, index ;
     while(str[i]){
        index = str[i] -  ' A ' ;
         if(p->next[index]==NULL) p->next[index]= new node() ;
        p = p->next[index] ;
        i ++ ;
    }
    p->count ++ ;
    p->id = ID ++ ;
}
void build_ac_automation(node *root){
     int i ;
    root->fail = NULL ;
    q[head++] = root ;
     while(head!=tail){
        node *temp = q[tail++] ;
        node *p = NULL ;
         for(i= 0; i< 30; i++){
             if(temp->next[i]!=NULL){
                 if(temp==root) temp->next[i]->fail = root ;
                 else{
                    p = temp->fail ;
                     while(p!=NULL){
                         if(p->next[i]!=NULL){
                            temp->next[i]->fail = p->next[i] ;
                             break ;
                        }
                        p = p->fail ;
                    }
                     if(p==NULL) temp->next[i]->fail = root ;
                }
                q[head++] = temp->next[i] ;
            }
        }
    }
}
void query(node *root){
     int i =  0, index ;
    node *p = root ;
     while(str[i]){
        index = str[i] -  ' A ' ;
         if(index< 0||index> 26)
            index =  27 ;
         while(p->next[index]==NULL && p!=root)
            p = p->fail ;
        p = p->next[index] ;
        p = (p==NULL) ? root : p ;
        node *temp = p ;
         while(temp!=root){
             if(temp->id!=- 1)
                vis[temp->id] ++ ;
            temp = temp->fail ;
        }
        i ++ ;
    }
}
int main(){
     int n ;
     while(~scanf( " %d ", &n)){
        memset(vis,  0,  sizeof(vis)) ;
        head = tail =  0 ;
        ID =  0 ;
        node *root =  new node() ;
         for( int i= 0; i<n; i++){
            getchar() ;
            scanf( " %s ", keyword[i]) ;
            insert(keyword[i], root) ;
        }
        build_ac_automation(root) ;
        getchar() ;
        scanf( " %s ", str) ;
        query(root) ;
         for( int i= 0; i<n; i++)
             if(vis[i])
                printf( " %s: %d\n ", keyword[i], vis[i]) ;
    }
     return  0 ;}