[HDU3065]病毒持续侵袭中（AC自动机）

传送门

AC自动机的又一模板，统计每个字符串在文本中的次数。

所以就不需要vis数组了。

——代码

#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;

int n, sz;
int ch[50001][130], val[50001], fail[50001], ans[50001];
char s[2000001], a[1001][50];

inline void insert(int k)
{
    int i, x, now = 0, len = strlen(a[k]);
    for(i = 0; i < len; i++)
    {
        x = a[k][i];
        if(!ch[now][x]) ch[now][x] = ++sz;
        now = ch[now][x];
    }
    val[now] = k;
}

inline void make_fail()
{
    int i, now = 0;
    queue <int> q;
    for(i = 0; i <= 128; i++)
     if(ch[0][i])
      q.push(ch[0][i]);
    while(!q.empty())
    {
        now = q.front(), q.pop();
        for(i = 0; i <= 128; i++)
        {
            if(!ch[now][i])
            {
                ch[now][i] = ch[fail[now]][i];
                continue;
            }
            fail[ch[now][i]] = ch[fail[now]][i];
            q.push(ch[now][i]);
        }
    }
}

inline void ac()
{
    int i, x, y, now = 0, len = strlen(s);
    for(i = 0; i < len; i++)
    {
        x = s[i];
        y = ch[now][x];
        while(y)
        {
            ans[val[y]]++;
            y = fail[y];
        }
        now = ch[now][x];
    }
}

int main()
{
    int i;
    while(~scanf("%d", &n))
    {
        sz = 0;
        memset(ch, 0, sizeof(ch));
        memset(val, 0, sizeof(val));
        memset(ans, 0, sizeof(ans));
        memset(fail, 0, sizeof(fail));
        for(i = 1; i <= n; i++)
        {
            scanf("%s", a[i]);
            insert(i);
        }
        make_fail();
        scanf("%s", s);
        ac();
        for(i = 1; i <= n; i++)
          if(ans[i])
            printf("%s: %d
", a[i], ans[i]);
    }
    return 0;
}