hdu2222（AC自动机）

题意是给了很多单词，然后给一个字符串，问字符串里包含了几个给定的单词

例如she he say shr her 问yasherhs含了几个，三个，注意she和he分别为两个单词，如果遍历一遍字符串显然是不够的，这时候用AC自动机就不会超时了

注意可能要有重复的单词

还有flag还是必须的

#include <iostream>

#include <string.h>

#include <iostream>

#include <queue>

#include <string.h>

#include <stdio.h>

#include <algorithm>

using namespacestd;

char str[1000005];

typedef struct node

{

   struct node *next[26],*fail;

   int num,flag;

}TrieNode,*trie;

TrieNode* creatTrie()

{

    TrieNode* node=(TrieNode*)malloc(sizeof(TrieNode));

   for(int i=0; i<26; i++)

        node->next[i]=NULL;

    node->fail=NULL;

    node->num=0;

    node->flag=0;

   return node;

}

void trieInsert(trie root)

{

   trie p=root;

   char *s=str;

   int id;

   while(*s)

    {

        //printf("%c ",*s);

        id=*s-'a';

       if(p->next[id]==NULL)

            p->next[id]=creatTrie();

        p=p->next[id];

        ++s;

    }

    p->num++;

}

void getfail(trie root)

{

   trie p=root,now,son;

    queue<trie>q;

    q.push(p);

   while(!q.empty())

    {

        now=q.front();

        q.pop();

       for(int i=0;i<26;i++)

        {

            son=now->next[i];

           if(son!=NULL)

            {

               if(now==root)

                    son->fail=root;

               else

                {

                    p=now->fail;

                   while(p)

                    {

                       if(p->next[i])

                        {

                            son->fail=p->next[i];

                           break;

                        }

                        p=p->fail;

                    }

                   if(!p)//

                        son->fail=root;

                }

                q.push(son);

            }

        }

    }

}

int find(trie root)

{

   trie p=root,temp;

    //ar *p=str;

   int t=0;

   int len=strlen(str);

   for(int i=0;i<len;i++)

    {

       int pos=str[i]-'a';

       while(!p->next[pos]&&p!=root)

        {

            p=p->fail;

        }

        

        p=p->next[pos];

       if(!p)

            p=root;

        temp=p;

        

       while(temp!=root&&temp->flag!=1)

        {

            t+=temp->num;

            temp->flag=1;

            temp=temp->fail;

        }

    }

   return t;

}

int main()

{

    

   int test,n;

   scanf("%d",&test);

   while(test--)

    {

       trie root=creatTrie();

       scanf("%d",&n);

       while(n--)

        {

           scanf("%s",str);

           trieInsert(root);

        }

       getfail(root);

       scanf("%s",str);

       printf("%d\n",find(root));

    }

    return0;

}

/*

2

5

she

se

say

shr

her

yasherhs

*/