diff options
Diffstat (limited to 'trie_prefix.cpp')
-rw-r--r-- | trie_prefix.cpp | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/trie_prefix.cpp b/trie_prefix.cpp new file mode 100644 index 0000000..b56e38b --- /dev/null +++ b/trie_prefix.cpp @@ -0,0 +1,45 @@ +// 208. Implement Trie (Prefix Tree) +#include "leetcode.h" + +class Trie { +public: + Trie() {} + void insert(string word) { + Trie *node = this; + for (char c : word) { + if (!node->next.count(c)) + node->next[c] = new Trie(); + node = node->next[c]; + } + node->is_word = true; + } + bool search(string word) { + Trie *node = this; + for (char c : word) { + if (!node->next.count(c)) + return false; + node = node->next[c]; + } + return node->is_word; + } + bool startsWith(string prefix) { + Trie *node = this; + for (char c : prefix) { + if (!node->next.count(c)) + return false; + node = node->next[c]; + } + return true; + } + +private: + map<char, Trie *> next = {}; + bool is_word = false; +}; + +int main() { + Trie *obj = new Trie(); + obj->insert("Trie"); + bool param_2 = obj->search("Trie"); + bool param_3 = obj->startsWith("Trie"); +} |