HDU 2896 病毒侵袭【AC自动机】

http://acm.hdu.edu.cn/showproblem.php?pid=2896
HDU 2896 病毒侵袭
大意：
有n种病毒，编码已知。有m个网站源码，求带病毒的网站个数以及其对应的病毒
Sample Input
3
aaa
bbb
ccc
2
aaabbbccc
bbaacc
 

Sample Output
web 1: 1 2 3
total: 1

hint:有3种病毒，编码分别为
aaa
bbb
ccc
2个网站，其中网站1包含了编号1,2,3病毒关键字

分析：
AC自动机。

PS:学长出的题，算是模板题吧。。。。贴上来，备着O(∩_∩)O~

View Code

1 #include < iostream >
2 #include < queue >
3 using namespace std;
4
5 // 子树节点是在插入时new的，
6 // 寻找失配指针中使用的队列是直接调用STL的
7 const int kind = 100 ; // 子树个数，因可见ASCII码为32-126，少于100个
8 bool visited[ 500 + 10 ]; // 记录当前病毒是否被访问
9 int ansid[ 4 ];
10 int ansnum;
11 struct node
12 {
13 node * fail;
14 node * next[kind];
15 int count; // 记录当前前缀是完整单词出现的个数
16 int id; // 记录病毒编号
17 node()
18 {
19 fail = NULL;
20 count = 0 ;
21 memset(next,NULL, sizeof (next));
22 }
23 }; // 寻找失配指针时需要用到的队列
24
25 char keyword[ 201 + 10 ]; // 关键字
26 char str[ 80000 ]; // 主串
27 void insert( char * str,node * root, int id)
28 {
29 node * p = root;
30 int i = 0 ,index;
31 while (str[i])
32 {
33 index = str[i] - 31 ;
34 if (p -> next[index] == NULL)
35 p -> next[index] = new node();
36
37 p = p -> next[index];
38 i ++ ;
39 }
40 p -> count ++ ;
41 p -> id = id;
42 }
43
44 // 寻找失败指针
45 void build_ac_automation(node * root)
46 {
47 int i;
48 queue < node *> Q;
49 root -> fail = NULL;
50 Q.push(root);
51 while ( ! Q.empty())
52 {
53 node * temp = Q.front(); // q[head++]; // 取队首元素
54 Q.pop();
55 node * p = NULL;
56 for (i = 0 ;i < kind;i ++ )
57 {
58 if (temp -> next[i] != NULL) // 寻找当前子树的失败指针
59 {
60 p = temp -> fail;
61 while (p != NULL)
62 {
63 if (p -> next[i] != NULL) // 找到失败指针
64 {
65 temp -> next[i] -> fail = p -> next[i];
66 break ;
67 }
68 p = p -> fail;
69 }
70
71 if (p == NULL) // 无法获取，当前子树的失败指针为根
72 temp -> next[i] -> fail = root;
73
74 Q.push(temp -> next[i]);
75 }
76 }
77 }
78 }
79
80 // 询问str中包含n个关键字中多少种
81 void query(node * root)
82 {
83 int i = 0 ,cnt = 0 ,index,len;
84 ansnum = 0 ;
85 memset(visited, false , sizeof (visited));
86 len = strlen(str);
87 node * p = root;
88 while (str[i])
89 {
90 index = str[i] - 31 ;
91 while (p -> next[index] == NULL && p != root) // 失配
92 p = p -> fail;
93 p = p -> next[index];
94 if (p == NULL) // 失配指针为根
95 p = root;
96
97 node * temp = p;
98 while (temp != root) // 寻找到当前位置为止是否出现病毒关键字
99 {
100 if (temp -> count != 0 && visited[temp -> id] == false )
101 {
102 visited[temp -> id] = true ;
103 ansid[ansnum ++ ] = temp -> id;
104 if (ansnum == 3 ) return ;
105 }
106
107 temp = temp -> fail;
108 }
109 i ++ ;
110 }
111 }
112 int main()
113 {
114 int n,t;
115 while (scanf( " %d " , & n) != EOF)
116 {
117
118 node * root = new node();
119 getchar();
120 int id = 0 ;
121 while (n -- )
122 {
123
124 gets(keyword);
125 insert(keyword,root, ++ id);
126 }
127
128 // 求失败指针
129 build_ac_automation(root);
130 int m;
131 scanf( " %d " , & m);
132 id = 0 ;
133 int total = 0 ;
134 while (m -- )
135 {
136 scanf( " %s " ,str);
137 query(root);
138 id ++ ;
139 if (ansnum != 0 )
140 {
141 total ++ ;
142 printf( " web %d: " ,id);
143 sort(ansid,ansid + ansnum); // 不要忘记排序，错在这里就郁闷了o(>﹏<)o
144 for ( int i = 0 ;i < ansnum;i ++ )
145 printf( " %d " ,ansid[i]);
146 printf( " \n " );
147 }
148 }
149 printf( " total: %d\n " ,total);
150 }
151 return 0 ;
152 }