Update and rename Add and Search Word - Data structure design.java to Design Add and Search Words Data Structure.java

Author: varunu28

Medium/Add and Search Word - Data structure design.java

@@ -0,0 +1,59 @@
+class WordDictionary {
+
+  private final TrieNode root;
+
+  public WordDictionary() {
+    this.root = new TrieNode();
+  }
+
+  public void addWord(String word) {
+    TrieNode curr = root;
+    for (char c : word.toCharArray()) {
+      if (!curr.children.containsKey(c)) {
+        curr.children.put(c, new TrieNode());
+      }
+      curr = curr.children.get(c);
+    }
+    curr.isWord = true;
+  }
+
+  public boolean search(String word) {
+    return searchHelper(word, 0, root);
+  }
+
+  private boolean searchHelper(String word, int idx, TrieNode curr) {
+    if (idx == word.length()) {
+      return curr.isWord;
+    }
+    if (curr.children.containsKey(word.charAt(idx))) {
+      return searchHelper(word, idx + 1, curr.children.get(word.charAt(idx)));
+    }
+    if (word.charAt(idx) == '.') {
+      for (char c : curr.children.keySet()) {
+        if (searchHelper(word, idx + 1, curr.children.get(c))) {
+          return true;
+        }
+      }
+    }
+    return false;
+  }
+
+
+  private static class TrieNode {
+
+    private final Map<Character, TrieNode> children;
+    private boolean isWord;
+
+    public TrieNode() {
+      this.children = new HashMap<>();
+      this.isWord = false;
+    }
+  }
+}
+
+/**
+ * Your WordDictionary object will be instantiated and called as such:
+ * WordDictionary obj = new WordDictionary();
+ * obj.addWord(word);
+ * boolean param_2 = obj.search(word);
+ */