1. 程式人生 > >hdu2896 病毒肆虐【AC自動機】

hdu2896 病毒肆虐【AC自動機】

思路 網站源碼 eve 都是 ron efi 查詢 true str

病毒侵襲

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 37730 Accepted Submission(s): 8348


Problem Description 當太陽的光輝逐漸被月亮遮蔽,世界失去了光明,大地迎來最黑暗的時刻。。。。在這樣的時刻,人們卻異常興奮——我們能在有生之年看到500年一遇的世界奇觀,那是多麽幸福的事兒啊~~
但網路上總有那麽些網站,開始借著民眾的好奇心,打著介紹日食的旗號,大肆傳播病毒。小t不幸成為受害者之一。小t如此生氣,他決定要把世界上所有帶病毒的網站都找出來。當然,誰都知道這是不可能的。小t卻執意要完成這不能的任務,他說:“子子孫孫無窮匱也!”(愚公後繼有人了)。
萬事開頭難,小t收集了好多病毒的特征碼,又收集了一批詭異網站的源碼,他想知道這些網站中哪些是有病毒的,又是帶了怎樣的病毒呢?順便還想知道他到底收集了多少帶病毒的網站。這時候他卻不知道何從下手了。所以想請大家幫幫忙。小t又是個急性子哦,所以解決問題越快越好哦~~

Input 第一行,一個整數N(1<=N<=500),表示病毒特征碼的個數。
接下來N行,每行表示一個病毒特征碼,特征碼字符串長度在20—200之間。
每個病毒都有一個編號,依此為1—N。
不同編號的病毒特征碼不會相同。
在這之後一行,有一個整數M(1<=M<=1000),表示網站數。
接下來M行,每行表示一個網站源碼,源碼字符串長度在7000—10000之間。
每個網站都有一個編號,依此為1—M。
以上字符串中字符都是ASCII碼可見字符(不包括回車)。

Output 依次按如下格式輸出按網站編號從小到大輸出,帶病毒的網站編號和包含病毒編號,每行一個含毒網站信息。
web 網站編號: 病毒編號 病毒編號 …
冒號後有一個空格,病毒編號按從小到大排列,兩個病毒編號之間用一個空格隔開,如果一個網站包含病毒,病毒數不會超過3個。
最後一行輸出統計信息,如下格式
total: 帶病毒網站數
冒號後有一個空格。

Sample Input 3 aaa bbb ccc 2 aaabbbccc bbaacc

Sample Output web 1: 1 2 3 total: 1

Source 2009 Multi-University Training Contest 10 - Host by NIT

Recommend gaojie | We have carefully selected several similar problems for you: 3065 2243 2825 3341 3247

題意:

給定n個模式串,m個文本串。問每個文本串中有多少個模式串出現過,輸出他們的編號。最後輸出所有含有模式串的文本串個數。

思路:

暴力跑AC自動機。有一些坑點。

第一是沒看到說字符是ASC碼中的字符,剛開始就開了26.看題解說開256會MLE,開130左右就夠了。

第二是對每個文本串進行統計時,用了一個vis數組。每一次查詢一個文本串,vis應該要重新初始化。

第三是輸入的字符串中會有空格,要用gets

AC自動機的題目一定要註意對每個節點進行初始化,細節方面要考慮好。

好像今天有點浮躁,寫題目很草率啊。

  1 #include <iostream>
  2 #include <set>
  3 #include <cmath>
  4 #include <stdio.h>
  5 #include <cstring>
  6 #include <algorithm>
  7 #include <vector>
  8 #include <queue>
  9 #include <map>
 10 //#include <bits/stdc++.h>
 11 using namespace std;
 12 typedef long long LL;
 13 #define inf 0x7f7f7f7f
 14 
 15 int m, n;
 16 const int maxn = 555;
 17 const int maxlen = 500 * 200 + 50;
 18 
 19 struct tree{
 20     int fail;
 21     int son[130];
 22     int ed;
 23     bool vis;
 24 }AC[maxlen];
 25 int tot = 0;
 26 char s[maxlen];
 27 
 28 void build(char s[], int id)
 29 {
 30     int len = strlen(s);
 31     int now = 0;
 32     for(int i = 0; i < len; i++){
 33         if(AC[now].son[s[i]] == 0){
 34             AC[now].son[s[i]] = ++tot;
 35         }
 36         now = AC[now].son[s[i]];
 37     }
 38     AC[now].ed = id;
 39 }
 40 
 41 void get_fail()
 42 {
 43     queue<int>que;
 44     for(int i = 0; i < 130; i++){
 45         if(AC[0].son[i] != 0){
 46             AC[AC[0].son[i]].fail = 0;
 47             que.push(AC[0].son[i]);
 48         }
 49     }
 50     while(!que.empty()){
 51         int u = que.front();
 52         que.pop();
 53         for(int i = 0; i < 130; i++){
 54             if(AC[u].son[i] != 0){
 55                 AC[AC[u].son[i]].fail = AC[AC[u].fail].son[i];
 56                 que.push(AC[u].son[i]);
 57             }
 58             else{
 59                 AC[u].son[i] = AC[AC[u].fail].son[i];
 60             }
 61         }
 62     }
 63 }
 64 
 65 int ans[5], top = 0;
 66 int AC_query(char s[])
 67 {
 68     for(int i = 0; i <= tot; i++){
 69         AC[i].vis = false;
 70     }
 71     int len = strlen(s);
 72     int now = 0, cnt = 0;
 73     for(int i = 0; i < len; i++){
 74         now = AC[now].son[s[i]];
 75         for(int t = now; t; t = AC[t].fail){
 76             if(!AC[t].vis && AC[t].ed != 0){
 77                 ans[top] = AC[t].ed;
 78                 //cout<<ans[top]<<endl;
 79                 top++;
 80                 cnt++;
 81                 AC[t].vis = true;
 82                 if(cnt >= 3)return cnt;
 83             }
 84         }
 85     }
 86     return cnt;
 87 }
 88 
 89 int main()
 90 {
 91     while(scanf("%d", &n) != EOF){
 92         for(int i = 0; i <= tot; i++){
 93             AC[i].ed = 0;
 94             AC[i].fail = 0;
 95             AC[i].vis = false;
 96             for(int j = 0; j < 130; j++){
 97                 AC[i].son[j] = 0;
 98             }
 99         }
100         tot = 0;
101         for(int i = 1; i <= n; i++){
102             getchar();
103             gets(s);
104             build(s, i);
105         }
106         AC[0].fail = 0;
107         get_fail();
108         scanf("%d", &m);
109         int num = 0;
110         for(int i = 1; i <= m; i++){
111             getchar();
112             gets(s);
113             top = 0;
114             if(AC_query(s)){
115                 printf("web %d:", i);
116                 sort(ans, ans + top);
117                 for(int j = 0; j < top; j++){
118                     printf(" %d", ans[j]);
119                 }
120                 printf("\n");
121                 num++;
122             }
123         }
124 
125         printf("total: %d\n", num);
126     }
127 
128     return 0;
129 }

hdu2896 病毒肆虐【AC自動機】