[HDU2896]病毒侵袭（AC自动机）

传送门

题目中文描述，赞！

除了val记录id以外就是模板。

注意：每次数组都要清0.0

——代码

 #include <cstdio>
 #include <queue>
 #include <cstring>
 #include <algorithm>

 using namespace std;

 int n, m, sz, tot;
 int ch[][], val[], ans[], fail[];
 bool vis[];
 char s[];

 inline void insert(int id)
 {
     int i, x, now = , len = strlen(s);
     for(i = ; i < len; i++)
     {
         x = s[i];
         if(!ch[now][x]) ch[now][x] = ++sz;
         now = ch[now][x];
     }
     val[now] = id;
 }

 inline void make_fail()
 {
     int i, now;
     queue <int> q;
     for(i = ; i <= ; i++)
      if(ch[][i])
       q.push(ch[][i]);
     while(!q.empty())
     {
         now = q.front(), q.pop();
         for(i = ; i <= ; i++)
         {
             if(!ch[now][i])
             {
                 ch[now][i] = ch[fail[now]][i];
                 continue;
             }
             fail[ch[now][i]] = ch[fail[now]][i];
             q.push(ch[now][i]);
         }
     }
 }

 inline void ac()
 {
     int i, now = , x, y, len = strlen(s);
     for(i = ; i < len; i++)
     {
         vis[now] = ;
         x = s[i];
         y = ch[now][x];
         while(y && !vis[y])
         {
             vis[y] = ;
             if(val[y]) ans[++ans[]] = val[y];
             y = fail[y];
         }
         now = ch[now][x];
     }
 }

 int main()
 {
     int i, j;
     scanf("%d", &n);
     for(i = ; i <= n; i++)
     {
         scanf("%s", s);
         insert(i);
     }
     make_fail();
     scanf("%d", &m);
     for(i = ; i <= m; i++)
     {
         scanf("%s", s);
         memset(vis, , sizeof(vis));
         memset(ans, , sizeof(ans));
         ac();
         if(ans[])
         {
             printf("web %d:", i);
             sort(ans + , ans + ans[] + );
             for(j = ; j <= ans[]; j++) printf(" %d", ans[j]);
             printf("\n");
             tot++;
         }
     }
     printf("total: %d\n", tot);
     return ;
 }