implement-trie-prefix-tree solved

Author: hsskey

implement-trie-prefix-tree/hsskey.js

@@ -0,0 +1,51 @@
+class TrieNode {
+    constructor() {
+        this.children = {};
+        this.isEndOfWord = false;
+    }
+}
+
+var Trie = function() {
+    this.root = new TrieNode();
+};
+
+/** 
+ * @param {string} word
+ * @return {void}
+ */
+Trie.prototype.insert = function(word) {
+    let node = this.root;
+    for (const char of word) {
+        if (!node.children[char]) {
+            node.children[char] = new TrieNode();
+        }
+        node = node.children[char];
+    }
+    node.isEndOfWord = true;
+};
+
+/** 
+ * @param {string} word
+ * @return {boolean}
+ */
+Trie.prototype.search = function(word) {
+    let node = this.root;
+    for (const char of word) {
+        if (!node.children[char]) return false;
+        node = node.children[char];
+    }
+    return node.isEndOfWord;
+};
+
+/** 
+ * @param {string} prefix
+ * @return {boolean}
+ */
+Trie.prototype.startsWith = function(prefix) {
+    let node = this.root;
+    for (const char of prefix) {
+        if (!node.children[char]) return false;
+        node = node.children[char];
+    }
+    return true;
+};