HDU 2222 Keywords Search 【ac自动机】

HDU  2222  Keywords  Search

 

题面见链接。

题解：

第一次学 AC 自动机，感觉差不多掌握基础了吧。

这是一道入门模板题。就是给定文本串，和一些模式串，求模式串在文本串出现的个数。

AC自动机主要步骤:
1、先建一棵 trie 树，然后记录该点是否为某模式串的最后一位节点，且以这一个节点结尾的模式串的个数。

2、预处理  fail  指针。 x 的 fail 指针连的是  fa[x]  的  fail  的  nxt  。目的是为了保证  fail  指针连的节点的模式串是 以 x 结尾的模式串的后缀。这样才能统计答案。

3、开始 AC 自动机主过程：从根节点开始按照文本串一位一位往下匹配。找到一个节点就检查他的  fail  节点是否能够加入答案，如果可以则加入答案，并标记为“已搜过”，下次就不用再搜一遍，提高时间效率。

贴代码：

#include<bits/stdc++.h>
using namespace std;
const int N=500005;
int t,n,rt,tot,q[N];
struct node{
    int nxt[26];
    int sum,fail;
}e[N];
char s[N<<1];
inline void insert(char *s)
{
    rt=0;
    for (int i=0; s[i]; i++)
    {
        if (!e[rt].nxt[s[i]-'a'])
          e[rt].nxt[s[i]-'a']=++tot;
        rt=e[rt].nxt[s[i]-'a'];
    }
    e[rt].sum++;
}
inline void getfail()
{
    int head=0,tail=1;
    q[1]=0;
    while (head<tail)
    {
        int x=q[++head];
        for (int i=0; i<26; i++)
        {
            if (int y=e[x].nxt[i])
            {
                if (x==0) e[y].fail=0;
                else {
                    int tmp=e[x].fail;
                    while (tmp && e[tmp].nxt[i]==0)
                      tmp=e[tmp].fail;
                    e[y].fail=e[tmp].nxt[i];
                }
                q[++tail]=y;
            }
        }
    }
}
inline int acmatch(char *s)
{
    getfail();
    rt=0;
    int cnt=0;
    for (int i=0; s[i]; i++)
    {
        while (rt && e[rt].nxt[s[i]-'a']==0)
          rt=e[rt].fail;
        int y=rt=e[rt].nxt[s[i]-'a'];
        while (y && ~e[y].sum)
        {
            cnt+=e[y].sum;
            e[y].sum=-1;
            y=e[y].fail;
        }
    }
    return cnt;
}
int main()
{
    scanf("%d",&t);
    while (t--)
    {
        memset(e,0,sizeof(e)); tot=0;
        scanf("%d",&n);
        for (int i=1; i<=n; i++)
        {
            scanf("%s",s);
            insert(s);
        }
        scanf("%s",s);
        printf("%d
",acmatch(s));
    }
    return 0;
}

加油加油加油！！！fighting fighting fighting!!!