首页 > 代码库 > Trie树学习2
Trie树学习2
数组实现的Trie树 字符容量有限,可以使用链表实现更为大容量的Trie
#include <iostream> #include <cstdio> #include <string> #include <cstring> #include <vector> #include <map> #include <set> #include <algorithm> #include <cstdlib> #include <list> #include <cmath> using namespace std; #define sigma_size 26 #define MAX_LEN 200001 struct trie_node{ trie_node* next[sigma_size]; bool is_terminal; trie_node(){ memset(next, 0, sizeof(next)); is_terminal = false; } }; struct Trie{ trie_node* root; int size; int idx(char ch){ return ch - 'A'; } Trie(){ root = new trie_node(); size = 1; } void Insert(string str){ trie_node* cur = root; int len = str.length(); for(int i = 0; i < len; i++){ int ch = idx(str[i]); if(cur->next[ch] == NULL){ cur->next[ch] = new trie_node(); } cur = cur->next[ch]; } cur->is_terminal = true; size++; } bool Query(string str){ trie_node* cur = root; int len = str.length(); for(int i = 0; i < len; i++){ int ch = idx(str[i]); if(cur->next[ch] == NULL){ return false; } else cur = cur->next[ch]; } if(cur->is_terminal) return true; else return false; } };
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。