【hdu2222-Keywords Search】AC自动机基础裸题

http://acm.hust.edu.cn/vjudge/problem/16403

题意：给定n个单词，一个字符串，问字符串中出现了多少个单词。（若单词her，he，字符串aher中出现了两个单词）

题解：

每个单词末尾节点sum=1；
find的时候每个点都顺着fail往上跳，加上该节点的sum，然后将这个sum清了；
注意同一个单词出现多次只算一次。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<queue>
 using namespace std;

 const int N=,L=;
 char s[L];
 int num,n;
 struct node{
     int son[];
     int fail,cnt;
 }a[N*];
 queue<int> q;

 void clear(int x)
 {
     a[x].cnt=;
     a[x].fail=;
     memset(a[x].son,,sizeof(a[x].son));
 }

 void trie(char *c)
 {
     int l=strlen(c);
     int x=;
     for(int i=;i<l;i++)
     {
         int t=c[i]-'a'+;
         if(!a[x].son[t])
         {
             num++;
             clear(num);
             a[x].son[t]=num;
         }
         x=a[x].son[t];
     }
     a[x].cnt++;
 }

 void buildAC()
 {
     while(!q.empty()) q.pop();
     for(int i=;i<=;i++)
         if(a[].son[i]) q.push(a[].son[i]);
     while(!q.empty())
     {
         int x=q.front();q.pop();
         int fail=a[x].fail;
         for(int i=;i<=;i++)
         {
             int y=a[x].son[i];
             if(y)
             {
                 a[y].fail=a[fail].son[i];
                 q.push(y);
             }
             else a[x].son[i]=a[fail].son[i];
         }
     }
 }

 int find(char *c)
 {
     int l=strlen(c);
     int x=,ans=;
     for(int i=;i<l;i++)
     {
         int t=c[i]-'a'+;
         while(x && !a[x].son[t]) x=a[x].fail;
         x=a[x].son[t];
         int p=x;
         while(p && a[p].cnt!=-)
         {
             ans+=a[p].cnt;
             a[p].cnt=-;
             p=a[p].fail;
         }
     }
     return ans;
 }

 int main()
 {
     freopen("a.in","r",stdin);
     freopen("a.out","w",stdout);
     int T;
     scanf("%d",&T);
     while(T--)
     {
         scanf("%d",&n);
         num=;
         clear();
         for(int i=;i<=n;i++)
         {
             scanf("%s",s);
             trie(s);
         }
         buildAC();
         scanf("%s",s);
         printf("%d\n",find(s));
     }
     return ;
 }