杭电 hdu 3065 病毒侵袭持续中

相当纠结～～

/* THE PROGRAM IS MADE BY PYY */ /*----------------------------------------------------------------------------// Copyright (c) 2011 panyanyany All rights reserved. URL : http://acm.hdu.edu.cn/showproblem.php?pid=3065 Name : 3065 病毒侵袭持续中 Date : Sunday, August 21, 2011 Time Stage : one hour and a half Result: 4463980 2011-08-21 16:05:47 Accepted 3065 156MS 5900K 3738 B C++ pyy 4463875 2011-08-21 15:54:25 Wrong Answer 3065 187MS 5900K 3703 B C++ pyy 4463866 2011-08-21 15:53:38 Wrong Answer 3065 312MS 27400K 3716 B C++ pyy 4463803 2011-08-21 15:47:02 Runtime Error (ACCESS_VIOLATION) 3065 46MS 27328K 3716 B C++ pyy 4463790 2011-08-21 15:45:11 Runtime Error (ACCESS_VIOLATION) 3065 31MS 8264K 3680 B C++ pyy 4463772 2011-08-21 15:43:07 Runtime Error (ACCESS_VIOLATION) 3065 31MS 6288K 3670 B C++ pyy 4463765 2011-08-21 15:42:21 Runtime Error (ACCESS_VIOLATION) 3065 0MS 252K 3677 B C++ pyy 4463755 2011-08-21 15:41:44 Runtime Error (ACCESS_VIOLATION) 3065 31MS 5840K 3675 B C++ pyy 4463744 2011-08-21 15:41:03 Runtime Error (ACCESS_VIOLATION) 3065 31MS 5844K 3673 B C++ pyy 4463739 2011-08-21 15:40:09 Runtime Error (ACCESS_VIOLATION) 3065 31MS 5844K 3671 B C++ pyy 4463734 2011-08-21 15:39:38 Runtime Error (ACCESS_VIOLATION) 3065 31MS 5836K 3670 B C++ pyy 4463728 2011-08-21 15:39:01 Runtime Error (ACCESS_VIOLATION) 3065 31MS 5828K 3668 B C++ pyy Test Data: Review: 犯了很多低级的错误，也犯了一些严重的错误…… //----------------------------------------------------------------------------*/ #include <stdio.h> #include <string.h> #include <stdlib.h> #define max(a, b) (((a) > (b)) ? (a) : (b)) #define min(a, b) (((a) < (b)) ? (a) : (b)) #define SafeAccess(a, b) (0 <= (a) && (a) < n && 0 <= (b) && (b) < n) #define infinity 0x0f0f0f0f #define minus_inf 0x80808080 #define MAXSIZE 2000009 #define LESSMAX 55 #define CHARNUM 26 typedef struct tagNODE { int cnt, num ; struct tagNODE * fail, * child[CHARNUM] ; } NODE ; #define root stack[0] NODE * tmp, * tmpFail, * newNode, * parntNode, * childNode ; NODE * queue[LESSMAX * 1000], * stack[LESSMAX * 1000] ; int stkPtr ; // for stack int head, tial ; // for queue int n ; int count[1001] ; char pattn[1001][LESSMAX], model[MAXSIZE] ; void makeTrie (char * ptn, int num) { int i, j ; int len = strlen (ptn) ; tmp = root ; for (i = 0 ; i < len ; ++i) { j = ptn[i] - 'A' ; if (!tmp->child[j]) { newNode = (NODE *) calloc (1, sizeof (NODE)) ; stack[stkPtr++] = newNode ; tmp->child[j] = newNode ; } tmp = tmp->child[j] ; } tmp->num = num ; ++tmp->cnt ; } void makeFail () { int i, j ; head = tial = 0 ; for (i = 0 ; i < CHARNUM ; ++i) { if (root->child[i]) { root->child[i]->fail = root ; queue[tial++] = root->child[i] ; } } while (head < tial) { parntNode = queue[head++] ; for (i = 0 ; i < CHARNUM ; ++i) { if (childNode = parntNode->child[i]) { tmpFail = parntNode->fail ; while (tmpFail != root && !tmpFail->child[i]) tmpFail = tmpFail->fail ; // 上一步没有对root的孩子检查是否与childNode匹配 // 所以这一步这检查一下，不能这样： // (tmp == root) ? root : tmpFail->child[i] ; childNode->fail = (tmpFail->child[i]) ? tmpFail->child[i] : root ; // 最后一步，不要忘了 queue[tial++] = childNode ; } } } } void ACAutomation () { int i, j ; int len = strlen (model) ; memset (count, 0, sizeof (count)) ; tmp = root ; for (i = 0 ; i < len ; ++i) { if ('A' <= model[i] && model[i] <= 'Z') { j = model[i] - 'A' ; while (!tmp->child[j] && tmp != root) tmp = tmp->fail ; // root 的child中有没有可以匹配的项，目前还是不知道的 // 所以下面还要再判断一下, 如果tmp 的child 能匹配到， // 则无论tmp 是什么都可以直接用，否则就要它就是一定 // 是root。这一句由于有两种实现方法，在不看答案代码 // 的时候，我曾经错误地写成： // (tmp == root) ? root : tmp->child[j] // 实际上，当tmp == root 的时候，tmp->child[j] 可能 // 是能够匹配的，而错误的语句却仍然把root 赋给了tmp tmp = (tmp->child[j]) ? tmp->child[j] : root ; tmpFail = tmp ; while (tmpFail->cnt) { ++count[tmpFail->num] ; // tmpFail->cnt = 0 ; tmpFail = tmpFail->fail ; } } else { /* 这句一定要加，否则当出现这样的数据的时候，便会错误： ----------------------------------------------------------------------- 3 AA BB CC A%AB%BC^&CC 正确输出： CC: 1 错误输出： AA: 1 BB: 1 CC: 2 ----------------------------------------------------------------------- 为什么会这样呢？ 因为它会认为A%A是连续的！ 所以当出现非英文大写字符时，便要从根部重新开始匹配了！ 当然，不加这句也可以，前提是CHARNUM 要变成128 个！ 并且要在相应的地方进行修改，这样的话，由于child的数目 增多了，循环的负担也会相应变大，时间自然有所增加！ */ tmp = root ; } } } void recycle () { while (stkPtr) free (stack[--stkPtr]) ; } int main () { int i, j ; while (scanf ("%d", &n) != EOF) { stkPtr = 1 ; stack [0] = (NODE *) calloc (1, sizeof (NODE)) ; for (i = 1 ; i <= n ; ++i) { scanf ("%s", pattn[i]) ; getchar () ; makeTrie (pattn[i], i) ; } scanf ("%s", model) ; makeFail () ; ACAutomation () ; for (i = 1 ; i <= n ; ++i) { if (count[i]) { printf ("%s: %d\n", pattn[i], count[i]) ; } } recycle () ; } return 0 ; }