【ZOJ3228】白濑肆の完美算法教室补番作，0.7%达成 白濑肆×AC自动机

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <queue>

using namespace std;

/**
    就是一个AC自动机
    但是每个单词有两个状态，[0]是可以overlap，[1]是不能overlap时候的次数
    我们用last数组记录下每个不能overlap的单词出现的位置，如果
    当前位置 - last[word] + 1 > strlen(word)，那么证明该单词并未overlap，更新last数组，对应次数+1即可
    补番作，水题。
*/
const int MAX_ALPHABET = 27;
const int MAX_TEXT = 100010;
const int MAX_WORD = 7;
const int MAX_WORD_NUM = 100010;
struct node
{
    int id;
    int cnt;
    int len;
    node* fail;
    node* next[MAX_ALPHABET];
    node()
    {
        id = -1; //指定该单词出现的位置。
        cnt = 0;
        fail = NULL;
        len = 0;
        memset(next,0,sizeof(next));
    }
};
int cnt;
int loc[MAX_WORD_NUM];
int type[MAX_WORD_NUM];
int strcnt[MAX_WORD_NUM][2]; // 每个词儿两个状态
int last[MAX_WORD_NUM]; //每个词最后一次出现的地方。
queue<node*> q;
变量区结束///
int AC_insert(node* root,char* dest,int flag)
{
    node* tmp = root;
    int l = strlen(dest);
    for(int i = 0 ; i < l ; i++)
    {
        int k = dest[i]-'a';
        if( tmp->next[k] == NULL )
        {
            tmp->next[k] = new node();
        }
        tmp = tmp->next[k];
    }
    if( tmp->id == -1)
    {
        tmp->id = cnt;
        cnt++;
    }
    tmp->cnt = tmp->cnt | flag;
    tmp->len = l;
    return tmp->id;  //标明该单词在单词列表中第几个出现。。。
}
void AC_build(node* root)
{
    root->fail = NULL;
    q.push(root);
    while(!q.empty())
    {
        node* tmp = q.front();
        q.pop();
        node *p;
        for(int i = 0 ; i < 26 ; i++)
        {
            if(tmp->next[i])
            {
                if( tmp == root )
                {
                    tmp->next[i]->fail = root;
                }
                else
                {
                    p = tmp->fail;
                    while(p)
                    {
                        if( p->next[i] )
                        {
                            tmp->next[i]->fail = p->next[i];
                            break;
                        }
                        p = p->fail;
                    }
                    if( !p )
                        tmp->next[i]->fail = root;
                }
                q.push(tmp->next[i]);
            }
        }
    }
}
void AC_search(node* root,char* dest)
{
    node* tmp;
    node* p;
    p = root;
    for(int i = 0 ; dest[i] ; i++)
    {
        int ind = dest[i] - 'a';
        while( p->next[ind]==NULL && p != root)
            p = p->fail;
        p = p->next[ind];
        if( p == NULL )
            p = root;
        tmp = p;
        while( tmp != NULL )
        {
            if( tmp->id != -1 )
            {
                if( tmp->cnt & 1 )
                {
                    strcnt[tmp->id][0]++;
                }
                if( tmp->cnt & 2 )
                {
                    if( i - tmp->len + 1 > last[tmp->id] )
                    {
                        strcnt[tmp->id][1]++;
                        last[tmp->id] = i;
                    }
                }
            }
            tmp = tmp->fail;
        }
    }
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("B:\\acm\\SummerVacation\\String-I\\B.in","r",stdin);
    freopen("B:\\acm\\SummerVacation\\String-I\\B.out","w",stdout);
#endif
    char dest[MAX_TEXT];
    char word[MAX_WORD];
    int t = 1;
    while(scanf("%s",dest) != EOF)
    {
        memset(last,-1,sizeof(last));
        memset(strcnt,0,sizeof(strcnt));
        cnt = 1;
        node* root = new node();


        int N;
        scanf("%d",&N);
        for(int i = 0 ; i < N ; i++)
        {
            scanf("%d %s",&type[i],word);
            loc[i] = AC_insert(root,word,type[i]+1);
        }
        AC_build(root);
        AC_search(root,dest);
        printf("Case %d\n",t++);
        int ans;
        for(int i = 0 ; i < N ; i++)
        {
            ans = strcnt[loc[i]][type[i]];
            printf("%d\n",ans);
        }
        printf("\n");
    }
#ifndef ONLINE_JUDGE
    fclose(stdin);
    fclose(stdout);
#endif
    return 0;
}