HDU 1251 统计拼图 Trie解决问题的方法
基本上找到一个标准前缀的问题是,只需要insert和search它功能。
这里的主要变化是我n该记录方法,这里n国旗代表的不是叶节点,但是话的标志这条道路后的数字。
然后找到需要找到一个词的前缀,假如未发现,返回0,这个前缀单词为前缀,假设找到,直接返回n就是答案了。由于有n个单词经过了这条路径。
查找效率是常数。
使用静态分配空间的办法。
#include <stdio.h> #include <string.h> const int MAX_N = 500000; const int ARR_SIZE = 26; const int MAX_WORD = 11; struct Node { int n; Node *arr[ARR_SIZE]; }; Node pool[MAX_N]; int poolId; void clearNode(Node *p) { p->n = 0; for (int i = 0; i < ARR_SIZE; i++) { p->arr[i] = NULL; } } void insert(Node *trie, char w[]) { int len = strlen(w); for (int i = 0; i < len; i++) { int j = w[i] - 'a'; if (trie->arr[j] == NULL) { trie->arr[j] = &pool[poolId++]; clearNode(trie->arr[j]); } trie = trie->arr[j]; trie->n++; //改动成能够高速查找前缀 } } int searchPre(Node *trie, char w[]) { int len = strlen(w); for (int i = 0; i < len; i++) { int j = w[i] - 'a'; if (!trie->arr[j]) return 0; trie = trie->arr[j]; } return trie->n; } int main() { Node *trie = &pool[0]; clearNode(trie); poolId = 1; char word[MAX_WORD]; while (gets(word) && word[0] != '\0') { insert(trie, word); } while (gets(word)) { printf("%d\n", searchPre(trie, word)); } return 0; }
版权声明:笔者靖心脏,景空间地址:http://blog.csdn.net/kenden23/,只有经过作者同意转载。