hdu 2222 Keywords Search (ac自动机入门题)

题意：

在目标串种找出单词出现的种类数。

题解：

直接搞。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const ll MOD=10007;
const int maxn=500005;
char str[1000005];


struct AC
{
    int next[500005][26],fail[500005],end[500005],Q[500005];
    int root,cnt;

    void Init()
    {
        cnt=0;
        root=newNode();
    }

    int newNode()
    {
        for(int i=0;i<26;i++)
            next[cnt][i]=-1;
        end[cnt++]=0;
        return cnt-1;
    }

    void Insert(char buff[])
    {
        int now=root;
        int len=strlen(buff);
        for(int i=0,k;i<len;i++)
        {
            k=buff[i]-'a';
            if(next[now][k]==-1)
                next[now][k]=newNode();
            now=next[now][k];
        }
        end[now]++;///计算这种单词的数量
    }

    void build()
    {
        int front,rear;
        front=rear=0;
        int now=root;
        for(int i=0;i<26;i++)
        {
            if(next[now][i]==-1)
                next[now][i]=root;
            else
            {
                fail[next[now][i]]=root;
                Q[rear++]=next[now][i];
            }
        }
        while(front<rear)
        {
            now=Q[front++];
            for(int i=0;i<26;i++)
            {
                if(next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];///同下一样，只不过因为下个是空所以只接用next[now][i]，非控就用fail来指向
                else
                {
                    fail[next[now][i]]=next[fail[now]][i];///适配只想fail指针对应节点的相同字母处，注意是相同字母处
                    Q[rear++]=next[now][i];
                }
            }
        }
    }

    int Search(char buff[])
    {
        int now=root,res=0;
        int len=strlen(buff);
        for(int i=0;i<len;i++)
        {
            now=next[now][buff[i]-'a'];
            int temp=now;
            while(temp!=root)///扫描
            {
                res+=end[temp];
                end[temp]=0;///防止重复计算
                temp=fail[temp];
            }
        }
        return res;
    }

    void Debug()
    {
        for(int i=0;i<cnt;i++)
        {
            printf("id=%3d fail=%3d end=%3d child=[",i,fail[i],end[i]);
            for(int j=0;j<26;j++)
                printf(" %3d",next[i][j]);
            printf(" ]\n");
        }
    }
};
AC ac;

int main()
{
    int T,n;
    scanf("%d",&T);
    while(T--)
    {
        ac.Init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%s",str);
            ac.Insert(str);
        }
        ac.build();
        scanf("%s",str);
        printf("%d\n",ac.Search(str));
    }
    return 0;
}