HDU 2896 病毒侵袭 AC自动机

题意：给你n个模式串，m个文本串，问你m个文本串中分别出现了几个模式串
解题思路：AC自动机模板题，我们只需要把 end 变为 n的序号就行。但是要注意这里所有串的范围
解题代码：
  1 // File Name: temp.cpp
  2 // Author: darkdream
  3 // Created Time: 2014年09月11日 星期四 15时18分128秒
  4 
  5 #include<vector>
  6 #include<list>
  7 #include<map>
  8 #include<set>
  9 #include<deque>
 10 #include<stack>
 11 #include<bitset>
 12 #include<algorithm>
 13 #include<functional>
 14 #include<numeric>
 15 #include<utility>
 16 #include<sstream>
 17 #include<iostream>
 18 #include<iomanip>
 19 #include<cstdio>
 20 #include<cmath>
 21 #include<cstdlib>
 22 #include<cstring>
 23 #include<ctime>
 24 #include<queue>
 25 #define LL long long
 26 #define maxn 100010
 27 using namespace std;
 28 vector<int> ans;
 29 struct Trie
 30 {
 31     int next[maxn][128],fail[maxn],end[maxn];
 32     int root, L;
 33     int newnode()
 34     {
 35         for(int i = 30 ;i < 127;i ++)
 36             next[L][i] = -1;
 37         end[L++] = 0 ;
 38         return L-1;
 39     }
 40     int init()
 41     {
 42         int L = 0 ; 
 43         root = newnode();
 44     }
 45     int insert(char buf[],int K)
 46     {
 47         int len = strlen(buf);
 48         int now = root;
 49         for(int i = 0 ;i < len ;i ++)
 50         {
 51             if(next[now][buf[i]] ==  -1)
 52             {
 53                 next[now][buf[i]] = newnode();
 54             }
 55             now = next[now][buf[i]];
 56         }
 57         end[now] = K;
 58     }
 59     void build()
 60     {
 61         queue<int> Q; 
 62         //这里为何要用优先队列来解决这个问题，主要原因是fail指针是按层数来的。
 63         fail[root] = root; 
 64         for(int i = 30;i < 127;i ++)
 65         {
 66             if(next[root][i] == -1)
 67             {
 68                 next[root][i] = root;  //指向root 是没有问题的，我们主要是通过end 数组对个数进行计数的。    
 69             }else{
 70                 fail[next[root][i]] = root;
 71                 Q.push(next[root][i]);
 72             }
 73         }
 74         while(!Q.empty())
 75         {
 76             int now = Q.front();
 77             Q.pop();
 78             for(int i = 30 ;i < 127;i ++)
 79             {
 80                 if(next[now][i] == -1)
 81                     next[now][i] =  next[fail[now]][i]; 
 82                 else{
 83                     fail[next[now][i]] = next[fail[now]][i];
 84                     Q.push(next[now][i]);
 85                 }
 86             }
 87         }
 88     }
 89     void query(char buf[])
 90     {
 91         int len = strlen(buf);
 92         int now = root ; 
 93         for(int i = 0 ;i < len;i ++)
 94         {
 95             now = next[now][buf[i]];
 96             int temp = now ; 
 97             while(temp != root)
 98             {
 99                 if(end[temp])
100                     ans.push_back(end[temp]);    
101                 temp = fail[temp];
102             }
103         }
104     }
105 };
106 char buf[10010];
107 Trie ac;
108 int main(){
109     int n;
110     while(scanf("%d",&n) != EOF)
111     {
112         ac.init();
113         getchar();
114         for(int i = 1 ;i <= n;i ++)
115         {
116             scanf("%[^\n]",buf);
117             getchar();
118             ac.insert(buf,i);
119         }
120         ac.build();
121         int m;
122         scanf("%d",&m);
123             getchar();
124         int sum = 0 ; 
125         for(int i = 1;i <= m;i ++)
126         {
127             scanf("%s",buf);
128             getchar();
129             //printf("***");
130             ans.clear();
131             ac.query(buf);
132             if(ans.size() != 0 )
133             {
134                 sum ++;
135             }else{
136               continue;
137             }
138             sort(ans.begin(),ans.end());
139             printf("web %d: ",i);
140             for(int j = 0 ;j < ans.size();j ++)
141                 printf(j == 0 ?"%d":" %d",ans[j]);
142             printf("\n");
143         }
144         printf("total: %d\n",sum);
145     }
146     return 0;
147 }
View Code
posted on 2014-09-11 17:24 dark_dream 阅读(165) 评论(0) 编辑收藏举报