HDU 2222 AC自动机

题意：

给出一些单词，和一段文本，输出文本中出现的单词个数，出现多次算一次。

模板题。

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>

#define N 1000010

using namespace std;

struct TR
{
    int son[26],fail,bot;
}tr[N];

int n,len,q[N<<2],num,ans;
bool vis[N];
char s[N],str[N];

inline void init()
{
    for(int i=0;i<26;i++) tr[0].son[i]=-1;
    tr[0].bot=tr[0].fail=0;
    num=0;
}

inline void insert()
{
    int now=0;
    len=strlen(s+1);
    for(int i=1;i<=len;i++)
    {
        if(tr[now].son[s[i]-'a']==-1)
        {
            num++;
            for(int j=0;j<26;j++) tr[num].son[j]=-1;
            tr[num].bot=tr[num].fail=0;
            tr[now].son[s[i]-'a']=num;
        }
        now=tr[now].son[s[i]-'a'];
    }
    tr[now].bot++;
}

inline void read()
{
    init();
    ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        insert();
    }
    for(int i=0;i<=num;i++) vis[i]=0;
    scanf("%s",str+1);
}

inline void bfs()
{
    int h=1,t=2,now,to,tmp;
    q[1]=0;
    while(h<t)
    {
        now=q[h++];
        for(int i=0;i<26;i++)
            if(tr[now].son[i]!=-1)
            {
                q[t++]=to=tr[now].son[i]; 
                tmp=now;
                while(tmp)
                {
                    tmp=tr[tmp].fail;
                    if(tr[tmp].son[i]!=-1)
                    {
                        tr[to].fail=tr[tmp].son[i];
                        break;
                    }
                }
            }
    }
}

inline void find(int x)
{
    while(x&&!vis[x])
    {
        vis[x]=true;
        ans+=tr[x].bot;
        x=tr[x].fail;
    }
}

inline void ac_automation()
{
    int now=0;
    len=strlen(str+1);
    for(int i=1;i<=len;i++)
    {
        if(tr[now].son[str[i]-'a']!=-1)
        {
            now=tr[now].son[str[i]-'a'];
            find(now);
        }
        else
        {
            int tmp=now;
            while(now)
            {
                now=tr[now].fail;
                if(tr[now].son[str[i]-'a']!=-1)
                {
                    now=tr[now].son[str[i]-'a'];
                    break;
                }
            }
            tr[tmp].son[str[i]-'a']=now;
            find(now);
        }
    }
}

inline void go()
{
    bfs();
    ac_automation();
    printf("%d\n",ans);
}

int main()
{
    int cas; scanf("%d",&cas);
    while(cas--) read(),go();
    return 0;
}