HDU 2222 Keywords Search（AC自动机模板题）

http://acm.hdu.edu.cn/showproblem.php?pid=2222

题意：
给出多个单词，最后再给出一个模式串，求在该模式串中包含了多少个单词。

思路：

AC自动机的模板题。

 #include<iostream>
 #include<algorithm>
 #include<cstring>
 #include<cstdio>
 #include<vector>
 #include<stack>
 #include<queue>
 #include<cmath>
 #include<map>
 #include<set>
 using namespace std;
 typedef long long ll;
 const int INF = 0x3f3f3f3f;
 const int maxn=+;

 int n;
 int num;

 struct Trie
 {
     int son[];
     int cnt;
     int fail;
 }t[*maxn];

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

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

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

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

 int main()
 {
     //freopen("in.txt","r",stdin);
     int T;
     char s[];
     scanf("%d",&T);
     while(T--)
     {
         num=;
         init();
         scanf("%d",&n);
         for(int i=;i<n;i++)
         {
             scanf("%s",s);
             trie(s);
         }
         buildAC();
         scanf("%s",s);
         printf("%d\n",query(s));
     }
     return ;
 }