HDU 2222 Keywords Search【AC自动机】

题目链接

题目意思

给你几个字符串，接着给你一个文本串，问你文本串中有几个前边所给的字符串。

解题思路

这就是一道简单的AC自动机的模板题，并没有什么难点。

代码部分

#include <iostream>#include <stdio.h>#include <stdlib.h>#include <queue>using namespace std;const int allson=26;char patten[60];///模式串char text[1000010];///文本串int ans;struct TrieNode{    struct TrieNode *son[allson];///儿子们    struct TrieNode *fail;///匹配失败时的指针方向    int num;}*root;///创建节点TrieNode* createNode(){    TrieNode *p;    p=(TrieNode*)malloc(sizeof(TrieNode));    for(int i=0; i<allson; i++)        p->son[i]=NULL;    p->num=0;    p->fail=NULL;    return p;}///插入模式串，构建字典树void insertPatten(){    TrieNode *p;    p=root;    int index=0;    while(patten[index]!='\0')    {        int lowercase=patten[index]-'a';        if(p->son[lowercase]==NULL)        {            p->son[lowercase]=createNode();        }        p=p->son[lowercase];        index++;    }    p->num++;}///求fail指针void build_AC_automaton(){    TrieNode *p;    p=root;    queue<TrieNode*>Q;    Q.push(p);    while(!Q.empty())    {        p=Q.front();        Q.pop();        for(int i=0; i<allson; i++)        {            if(p->son[i]!=NULL)            {                if(p==root)                {                    p->son[i]->fail=root;                }                else                {                    TrieNode *node=p->fail;                    while(node!=NULL)                    {                        if(node->son[i]!=NULL)                        {                            p->son[i]->fail=node->son[i];                            break;                        }                        node=node->fail;                    }                    if(node==NULL)                        p->son[i]->fail=root;                }                Q.push(p->son[i]);            }        }    }}void find_in_AC_automaton(){    TrieNode *p;    p=root;    int index=0;    while(text[index]!='\0')    {        int lowercase=text[index]-'a';        while(p->son[lowercase]==NULL&&p!=root)            p=p->fail;        p=p->son[lowercase];        if(p==NULL)            p=root;        TrieNode *temp=p;        while(temp!=NULL&&temp->num!=-1)        {            ans+=temp->num;            temp->num=-1;            temp=temp->fail;        }        index++;    }}void freeNode(TrieNode *node){    if(node!=NULL)    {        for(int i=0; i<allson; i++)        {            freeNode(node->son[i]);        }    }    free(node);}int main(){    int t,n;    scanf("%d",&t);    while(t--)    {        scanf("%d",&n);        root=createNode();        for(int i=0; i<n; i++)        {            scanf("%s",patten);            insertPatten();        }        scanf("%s",text);        build_AC_automaton();        ans=0;        find_in_AC_automaton();        printf("%d\n",ans);        freeNode(root);    }    return 0;}