ZOJ3228 Searching the String AC自动机

题意：给你一个文本串，其中模式串有两种模式，可以重复和不可以重复，分别有多少个模式串

解题思路：

在  Trie 里面多加几维数组来维护 重复和不重复的和，由于不够优美，差点超内存。

解题代码：

// File Name: temp.cpp
// Author: darkdream
// Created Time: 2014年09月11日 星期四 15时18分4秒

#include<vector>
#include<list>
#include<map>
#include<set>
#include<deque>
#include<stack>
#include<bitset>
#include<algorithm>
#include<functional>
#include<numeric>
#include<utility>
#include<sstream>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<ctime>
#include<queue>
#define LL long long
#define maxn 100105*7
using namespace std;
int hs[100105];
struct Trie
{
    int next[maxn][26],fail[maxn],end[maxn],ansend0[maxn],ansend1[maxn],first[maxn];
    vector <int> exend0[maxn];
    vector <int> exend1[maxn];
    int root, L;
    int newnode()
    {
        memset(next[L],-1,sizeof(next[L]));
        exend0[L].clear();
        exend1[L].clear();
        ansend0[L] =0 ;
        ansend1[L] = 0 ;
        end[L++] = 0 ;
        return L-1;
    }
    void init()
    {
        L = 0 ; 
        root = newnode();
    }
    void insert(char buf[] ,int k ,int status)
    {
        int now = root;
        int len = strlen(buf);
        for(int i = 0 ;i < len ;i ++)
        {
            if(next[now][buf[i] - 'a'] ==  -1)
            {
                next[now][buf[i] - 'a'] = newnode();
            }
            now = next[now][buf[i]- 'a'];
        }
        end[now] = len;
        if(status == 0)
        {
          exend0[now].push_back(k);
        }else{
          exend1[now].push_back(k);
        }
    }
    void build()
    {
        queue<int> Q;
        fail[root] = root; 
        for(int i = 0;i < 26;i ++)
        {
            if(next[root][i] == -1)
            {
                next[root][i] = root;  //指向root 是没有问题的，我们主要是通过end 数组对个数进行计数的。    
            }else{
                fail[next[root][i]] = root;
                Q.push(next[root][i]);
            }
        }
        while(!Q.empty())
        {
            int now = Q.front();
            Q.pop();
            for(int i = 0 ;i < 26;i ++)
            {
                if(next[now][i] == -1)
                {
                    next[now][i] =  next[fail[now]][i]; 
                }
                else{
                    fail[next[now][i]] = next[fail[now]][i];
                    Q.push(next[now][i]);
                }
            }
        }
    }
    void query(char buf[])
    {
        memset(first,-1,sizeof(first));
        memset(hs,0,sizeof(hs));
        int len = strlen(buf);
        int now = root;
        for(int i = 0 ;i < len;i ++)
        {
            now = next[now][buf[i] - 'a'];
            int temp = now ; 
            while(temp != root)
            {
               if(end[temp])
               {
                   if(first[temp] <= i - end[temp])
                   {
                      ansend1[temp] ++ ;
                      first[temp] = i ; 
                   }
                   ansend0[temp] ++ ; 
               }
               temp = fail[temp];
            }

        }
        for(int i = 0 ;i < L ;i ++)
        {
            if(end[i])
            {
              int t = exend0[i].size();
              for(int j = 0 ;j < t ;j ++)
              {
                  hs[exend0[i][j]] = ansend0[i];
              }
              t = exend1[i].size();
              for(int j = 0 ;j < t ;j ++)
              {
                  hs[exend1[i][j]] = ansend1[i];
              }
            
            }
        }
    }
};
char buf[101005];
char temp[10];
Trie ac;
int main(){
    int n;
    int ca= 0 ;
    //freopen("in","r",stdin);
    while(scanf("%s",buf) != EOF)
    {
        ca ++ ;
        scanf("%d",&n);
        ac.init();
        int status;
        for(int i =1 ;i <= n;i ++)
        {
            scanf("%d %s",&status,temp);
            ac.insert(temp,i,status);
        }
        ac.build();
        ac.query(buf);
        printf("Case %d
",ca);
        for(int i = 1;i <= n;i ++)
            printf("%d
",hs[i]);
        printf("
");
    }
    return 0;
}