AC自动机 - 多模式串的匹配运用 --- HDU 2896

// Memory   Time
// 1347K     0MS
// by : Snarl_jsb
// 2014-09-30-11.01
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<climits>
#include<cmath>
#define LL long long
using namespace std;


const int N = 10010;
char str [N ];
struct node
{
    node * next [ 128 ];     //  每个结点都对应128个字母的指针
    node * fail;     //      失配指针
    int count;       //
    int num;
    node()       //  构造函数初始化
    {
        for( int i = 0; i < 128; i ++)
            next [ i ] = NULL;
        count = 0;
        fail = NULL;
        num = 0;
    }
} * q [ 50 *N ];
node * root;
int head , tail;

void Insert( char * str , int num) //   插入单词.相当于构建一个Trie树
{
    node *p = root;
    int i = 0 , index;
    while( str [ i ]) {
        index = str [ i ] - 31; //  转化为相对数字来存
        if(p -> next [ index ] == NULL) // 该字母未插入过
           p -> next [ index ] = new node();     //  为该字母申请一个结点
       p = p -> next [ index ];     //   移至下一个
        i ++;
    }
   p -> count ++;     //      记录该结点的单词总共插入的次数
   p -> num = num;
}
void build_ac_automation( node * root)         //      bfs建立fail指针
{
    root -> fail = NULL;
    q [ tail ++ ] = root;
    while( head < tail) {
        node * temp = q [ head ++ ];
        node *p = NULL;
        for( int i = 0; i < 128; 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 [ tail ++ ] = temp -> next [ i ];
            }
        }
    }
}
int total = 0;
int Query( node * root , int num)       //  匹配 + 统计
{
    int i = 0 , cnt = 0 , index;
    node *p = root;
    int idx = 0;
    int ans [ 100 ];

    while( str [ i ])
    {
        index = str [ i ] - 31;
        while(p -> next [ index ] == NULL && p != root) //前缀是相同的，所以不管哪个指针走到了count不为0的结点上，那么该结点所代表的单词就匹配成功
           p = p -> fail; //失配情况下，p指针指向p->fail.(相当于KMP的next数组)
       p = p -> next [ index ]; //由于现在所在的位置是父节点，所以需要向下移动一个位置
        if(p == NULL)
           p = root; //如果匹配失败，移动到root，重新开始匹配
        node * temp = p; //
        while( temp != root && temp -> count != - 1)   //统计--如果匹配成功,那么count>1，表示该结点代表的单词数量;否则表示该结点没有单词
        {
            if( temp -> count > 0)
            {
                cnt ++; //统计该单词出现的次数
                ans [ ++ idx ] = temp -> num;
            }
//            temp->count = -1;   //标记为-1，表示该单词已经加入了cnt中，下次就不用重复统计
            temp = temp -> fail; //判断整条链上的匹配情况
        }
        i ++;
    }
    if( idx == 0)
        return 0;
    printf( "web %d: " , num);
    sort( ans + 1 , ans + 1 + idx);
    total ++;
    for( int i = 1; i < idx; ++ i)
    {
        printf( "%d " , ans [ i ]);
    }
    printf( "%d \n " , ans [ idx ]);
    return cnt;
}

int main()
{
    int n;
    cin >>n;
    head = tail = 0;
    root = new node();
    for( int i = 1; i <=n; ++ i)
    {
        scanf( "%s" , str);
        Insert( str , i);
    }
    build_ac_automation( root);       //  建树
    int m;
    cin >> m;
    total = 0;
    for( int i = 1; i <= m; ++ i)
    {
        scanf( "%s" , str);
        Query( root , i);
    }
    printf( "total: %d \n " , total);
    return 0;
}