hdu 2222 keywords search

https://vjudge.net/problem/HDU-2222

题意：

给出若干个模式串，然后给出文本串，问这些模式串在文本串中出现了几次。

思路：

妥妥的ac自动机，没有什么好说的，直接上模板，不过要注意maxn的大小是跟文本串的长度一样的。

代码：

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;

const int maxn = 1000005;

struct ac_auto
{
    int chd[maxn][26],v[maxn],f[maxn],last[maxn],sz,ans;

    void init()
    {
        sz = 1;
        ans = 0;

        memset(chd[0],0,sizeof(chd[0]));
        memset(f,0,sizeof(f));
        memset(v,0,sizeof(v));
    }

    void insert(char *p)
    {
        int cur = 0;
        for (;*p;p++)
        {
            if (!chd[cur][*p-'a'])
            {
                memset(chd[sz],0,sizeof(chd[sz]));
                chd[cur][*p-'a'] = sz++;
            }
            cur = chd[cur][*p-'a'];
        }
        v[cur]++;
    }

    bool query(char *p)
    {
        int cur = 0;

        for (;*p;p++)
        {
            if (!chd[cur][*p-'a']) break;
            cur = chd[cur][*p-'a'];
        }

        return v[cur] && (!(*p));
    }

    void getfail()
    {
        queue<int> q;

        f[0] = 0;

        for (int c = 0;c < 26;c++)
        {
            int u = chd[0][c];

            if (u)
            {
                f[u] = 0;q.push(u);
                last[u] = 0;
            }
        }

        while (!q.empty())
        {
            int r = q.front();q.pop();

            for (int c = 0;c < 26;c++)
            {
                int u = chd[r][c];

                if (!u)
                {
                    chd[r][c] = chd[f[r]][c];
                    continue;
                }

                q.push(u);

                int vv = f[r];

                while (vv&&!chd[vv][c]) vv = f[vv];
                f[u] = chd[vv][c];
                last[u] = v[f[u]] ? f[u] : last[f[u]];
            }
        }
    }

    void solve(int j)
    {
        if (!j) return;
        if (v[j])
        {
            ans += v[j];
            v[j] = 0;
        }

        solve(last[j]);
    }

    void find(char *t)
    {
        int n = strlen(t),j = 0;
        getfail();

        for (int i = 0;i < n;i++)
        {
            j = chd[j][t[i]-'a'];

            if (v[j]) solve(j);
            else if (last[j]) solve(last[j]);
        }
    }
}ac;

char tt[1000005];

int main()
{
    int t;

    scanf("%d",&t);

    while (t--)
    {
        ac.init();

        int n;

        scanf("%d",&n);

        for (int i = 0;i < n;i++)
        {
            char tmp[60];

            scanf("%s",tmp);

            ac.insert(tmp);
        }

        scanf("%s",tt);

        ac.find(tt);

        printf("%d
",ac.ans);
    }

    return 0;
}