[HDU2222]Keywords Search（AC自动机）

Keywords Search

一道模板题，但对于我这种初学者来说也是不好做的。

对于AC自动机的理解，本蒟蒻暂时还理解不好，不多说了。

看看这个人的blog。

——本题代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#define N 500005

using namespace std;

char s[N << 1];
int T, n, sz, ans;
int ch[N][26], val[N], fail[N];
bool vis[N];
queue <int> q;

inline void clear()
{
    n = sz = ans = 0;
    memset(ch, 0, sizeof(ch));
    memset(vis, 0, sizeof(vis));
    memset(val, 0, sizeof(val));
    memset(fail, 0, sizeof(fail));
}

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

inline void make_fail()
{
    int i, now;
    while(!q.empty()) q.pop();
    //第二层特殊处理，将第二层节点的 fail 指针指向 root（其实就是0，也就不用管） 
    for(i = 0; i < 26; i++) 
     if(ch[0][i])
      q.push(ch[0][i]);
    while(!q.empty())
    {
        now = q.front(), q.pop();
        for(i = 0; i < 26; 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 x, y, len = strlen(s), i, now = 0;
    for(i = 0; i < len; i++)
    {
        vis[now] = 1;
        x = s[i] - 'a';
        y = ch[now][x];
        while(y && !vis[y])
        {
            vis[y] = 1;
            ans += val[y];
            y = fail[y];
        }
        now = ch[now][x];
    }
}

int main()
{
    int i;
    scanf("%d", &T);
    while(T--)
    {
        clear();
        scanf("%d", &n);
        for(i = 1; i <= n; i++)
        {
            scanf("%s", s);
            insert();
        }
        scanf("%s", s);
        make_fail();
        ac();
        printf("%d
", ans);
    }
    return 0;
}