HDU 3065 病毒侵袭持续中（AC自动机）

　　这题数据太水，一开始没有加上Get的方法也能AC。。话说AC自动机中一定要注意加上Get的方法！（不然，同一个后缀的其他单词就没被算上了。）

　　代码如下：

 #include <stdio.h>
 #include <algorithm>
 #include <string.h>
 #include <queue>
 #include <string>
 #include <map>
 #include <iostream>
 using namespace std;
 const int MAX_N =  + ;
 const int MAX_Tot = * + ;

 int ans[+];
 map<int,string> M;

 struct Aho
 {
     struct state
     {
         int nxt[];
         int fail,cnt;
     }stateTable[MAX_Tot];

     int size;

     queue<int> que;

     void init()
     {
         while(que.size()) que.pop();
         for(int i=;i<MAX_Tot;i++)
         {
             memset(stateTable[i].nxt,,sizeof(stateTable[i].nxt));
             stateTable[i].fail = stateTable[i].cnt = ;
         }
         size = ;
     }

     void insert(char *s,int which)
     {
         int n = strlen(s);
         int now = ;
         for(int i=;i<n;i++)
         {
             char c = s[i];
             if(!stateTable[now].nxt[c-'A'])
                 stateTable[now].nxt[c-'A'] = size++;
             now = stateTable[now].nxt[c-'A'];
         }
         stateTable[now].cnt = which;
     }

     void build()
     {
         stateTable[].fail = -;
         que.push();

         while(que.size())
         {
             int u = que.front();que.pop();
             for(int i=;i<;i++)
             {
                 if(stateTable[u].nxt[i])
                 {
                     if(u == ) stateTable[stateTable[u].nxt[i]].fail = ;
                     else
                     {
                         int v = stateTable[u].fail;
                         while(v != -)
                         {
                             if(stateTable[v].nxt[i])
                             {
                                 stateTable[stateTable[u].nxt[i]].fail = stateTable[v].nxt[i];
                                 break;
                             }
                             v = stateTable[v].fail;
                         }
                         if(v == -) stateTable[stateTable[u].nxt[i]].fail = ;
                     }
                     que.push(stateTable[u].nxt[i]);
                 }
             }
         }
     }

     void Get(int u)
     {
         while(u)
         {
             ans[stateTable[u].cnt] ++;
             u = stateTable[u].fail;
         }
     }

     void match(char *s)
     {
         int n = strlen(s);
         int now = ;
         for(int i=;i<n;i++)
         {
             char c = s[i];
             if(c<'A' || c>'Z') {now = ;continue;}
             if(stateTable[now].nxt[c-'A']) now = stateTable[now].nxt[c-'A'];
             else
             {
                 int p = stateTable[now].fail;
                 while(p != - && stateTable[p].nxt[c-'A'] == ) p = stateTable[p].fail;
                 if(p == -) now = ;
                 else now = stateTable[p].nxt[c-'A'];
             }
             if(stateTable[now].cnt)
             {
                 Get(now);
                 //ans[stateTable[now].cnt] ++;
                 /*
                     一定要用Get这样的方法，
                     不然，同一个后缀的无法被算上了
                 */
             }
         }
     }
 }aho;

 int n;
 char s[MAX_N];

 int main()
 {
     while(scanf("%d",&n)==)
     {
         memset(ans,,sizeof(ans));
         M.clear();
         aho.init();
         for(int i=;i<=n;i++)
         {
             scanf("%s",s);
             M[i] = s;
             aho.insert(s,i);
         }
         aho.build();
         scanf("%s",s);
         aho.match(s);
         for(int i=;i<=n;i++)
         {
             if(ans[i])
             {
                 cout << M[i] << ": " << ans[i] << endl;
             }
         }
     }
 }

————————————————————————————————————————————

　　发现上面的代码有问题（虽然能AC），正确代码如下：

 #include <stdio.h>
 #include <algorithm>
 #include <string.h>
 #include <queue>
 #include <string>
 #include <map>
 #include <iostream>
 using namespace std;
 const int MAX_N =  + ;
 const int MAX_Tot = * + ;

 int ans[+];
 map<int,string> M;

 struct Aho
 {
     struct state
     {
         int nxt[];
         int fail,cnt;
     }stateTable[MAX_Tot];

     int size;

     queue<int> que;

     void init()
     {
         while(que.size()) que.pop();
         for(int i=;i<MAX_Tot;i++)
         {
             memset(stateTable[i].nxt,,sizeof(stateTable[i].nxt));
             stateTable[i].fail = stateTable[i].cnt = ;
         }
         size = ;
     }

     void insert(char *s,int which)
     {
         int n = strlen(s);
         int now = ;
         for(int i=;i<n;i++)
         {
             char c = s[i];
             if(!stateTable[now].nxt[c-'A'])
                 stateTable[now].nxt[c-'A'] = size++;
             now = stateTable[now].nxt[c-'A'];
         }
         stateTable[now].cnt = which;
     }

     void build()
     {
         stateTable[].fail = -;
         que.push();

         while(que.size())
         {
             int u = que.front();que.pop();
             for(int i=;i<;i++)
             {
                 if(stateTable[u].nxt[i])
                 {
                     if(u == ) stateTable[stateTable[u].nxt[i]].fail = ;
                     else
                     {
                         int v = stateTable[u].fail;
                         while(v != -)
                         {
                             if(stateTable[v].nxt[i])
                             {
                                 stateTable[stateTable[u].nxt[i]].fail = stateTable[v].nxt[i];
                                 break;
                             }
                             v = stateTable[v].fail;
                         }
                         if(v == -) stateTable[stateTable[u].nxt[i]].fail = ;
                     }
                     que.push(stateTable[u].nxt[i]);
                 }
             }
         }
     }

     void Get(int u)
     {
         while(u)
         {
             ans[stateTable[u].cnt] ++;
             u = stateTable[u].fail;
         }
     }

     void match(char *s)
     {
         int n = strlen(s);
         int now = ;
         for(int i=;i<n;i++)
         {
             char c = s[i];
             if(c<'A' || c>'Z') {now = ;continue;}
             if(stateTable[now].nxt[c-'A']) now = stateTable[now].nxt[c-'A'];
             else
             {
                 int p = stateTable[now].fail;
                 while(p != - && stateTable[p].nxt[c-'A'] == ) p = stateTable[p].fail;
                 if(p == -) now = ;
                 else now = stateTable[p].nxt[c-'A'];
             }
             //if(stateTable[now].cnt)
             {
                 Get(now);
             }
         }
     }
 }aho;

 int n;
 char s[MAX_N];

 int main()
 {
     while(scanf("%d",&n)==)
     {
         memset(ans,,sizeof(ans));
         M.clear();
         aho.init();
         for(int i=;i<=n;i++)
         {
             scanf("%s",s);
             M[i] = s;
             aho.insert(s,i);
         }
         aho.build();
         scanf("%s",s);
         aho.match(s);
         for(int i=;i<=n;i++)
         {
             if(ans[i])
             {
                 cout << M[i] << ": " << ans[i] << endl;
             }
         }
     }
 }
 /*
 3
 ABCDEFG
 ABCDE
 DEF
 ABCDEFG
 */