首页 > 代码库 > HDU 2896 病毒侵袭 AC自动机题解
HDU 2896 病毒侵袭 AC自动机题解
本题是在text里面查找key word的增强版,因为这里有多个text。
那么就不可以简单把Trie的叶子标志记录修改成-1进行加速了,可以使用其他技术,我直接使用个vis数组记录已经访问过的节点,达到加速效果,速度还算挺快的。
不过看discuss里面有人直接使用Trie,做出了140ms的速度,而且他的程序严格来说并不正确,可见本题的数据很水啊。Trie的时间效率肯定比AC自动机低,但是在数据很水的特殊情况下,Trie的速度也可以很快的。
注意两个细节:
1 病毒也需要安装顺序输出,不小心就遗漏了。这里害我错了几次。
2 还有就是题目的字符不确定是多少,使用33-94范围的字符测试答案正确的。
#include <stdio.h> #include <string.h> #include <queue> #include <set> #include <algorithm> using namespace std; const int MAX_N = 501; const int MAX_WORD = 201; const int MAX_M = 1001; const int MAX_TXT = 10001; const int MAX_V = 3; const int ARR_SIZE = 94; char virus[MAX_WORD], web[MAX_TXT]; int webNum[MAX_V]; int M, N; inline int getIndex(char ch) { return ch - 33; } struct Node { int n, num, id; Node *fail; Node *arr[ARR_SIZE]; }; void clearNode(Node *rt) { rt->n = 0; rt->num = 0; rt->id = 0; rt->fail = NULL; for (int i = 0; i < ARR_SIZE; i++) { rt->arr[i] = NULL; } } Node gPool[MAX_N*MAX_WORD], *Trie; int gPoolID; bool vis[MAX_N*MAX_WORD]; //怎么老是忘记使用visited技术的? void insertTrie(char *virus, int num) { Node *pCrawl = Trie; for (; *virus; virus++) { int id = getIndex(*virus); if (!pCrawl->arr[id]) { pCrawl->arr[id] = &gPool[gPoolID++]; clearNode(pCrawl->arr[id]); pCrawl->arr[id]->id = gPoolID-1; } pCrawl = pCrawl->arr[id]; } pCrawl->n++; pCrawl->num = num; } void buildFail() { queue<Node *> qu; qu.push(Trie); while (!qu.empty()) { Node *p = qu.front(); qu.pop(); for (int i = 0; i < ARR_SIZE; i++) { if (!p->arr[i]) continue; p->arr[i]->fail = Trie; Node *fail = p->fail; while (fail) { if (fail->arr[i]) { p->arr[i]->fail = fail->arr[i]; break; } fail = fail->fail; } qu.push(p->arr[i]); } } } int searchVirus(char *web) { int n = 0; Node *pCrawl = Trie; memset(vis, 0, sizeof(bool)*(gPoolID+1)); for (; *web; web++) { int i = getIndex(*web); while (!pCrawl->arr[i] && pCrawl != Trie) pCrawl = pCrawl->fail; if (pCrawl->arr[i]) { pCrawl = pCrawl->arr[i]; Node *p = pCrawl; while (p && !vis[p->id])//使用vis比直接检查数组值快捷方便 { if (p->n) webNum[n++] = p->num; vis[p->id] = true; p = p->fail; } if (n == MAX_V) break; //At most MAX_V virus one web } } return n; } int main() { Trie = &gPool[0]; while (scanf("%d", &N) != EOF) { gPoolID = 1; clearNode(Trie); getchar(); for (int i = 1; i <= N; i++) { gets(virus); insertTrie(virus, i); } buildFail(); scanf("%d", &M); getchar(); int cnt = 0; for (int i = 1; i <= M; i++) { gets(web); int n = searchVirus(web); if (n) { cnt++; printf("web %d:", i); sort(webNum, webNum + n);//Watch out: order! for (int j = 0; j < n; j++) { printf(" %d", webNum[j]); } putchar('\n'); } } printf("total: %d\n", cnt); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。