design-add-and-search-words-data-structure solved

Author: hsskey

design-add-and-search-words-data-structure/hsskey.js

@@ -0,0 +1,47 @@
+class TrieNode {
+    constructor() {
+        this.children = {};
+        this.word = false;
+    }
+}
+
+class WordDictionary {
+    constructor() {
+        this.root = new TrieNode();
+    }
+
+    addWord(word) {
+        let cur = this.root;
+        for (let c of word) {
+            if (!cur.children[c]) {
+                cur.children[c] = new TrieNode();
+            }
+            cur = cur.children[c];
+        }
+        cur.word = true;
+    }
+
+    search(word) {
+        const dfs = (j, root) => {
+            let cur = root;
+            for (let i = j; i < word.length; i++) {
+                const c = word[i];
+                if (c === '.') {
+                    for (let child of Object.values(cur.children)) {
+                        if (dfs(i + 1, child)) {
+                            return true;
+                        }
+                    }
+                    return false;
+                } else {
+                    if (!cur.children[c]) {
+                        return false;
+                    }
+                    cur = cur.children[c];
+                }
+            }
+            return cur.word;
+        };
+        return dfs(0, this.root);
+    }
+}