Day 5 - Add and Search Word - Data structure design

Author: arshad115

5-add-and-search-word-data-structure-design.py

@@ -0,0 +1,55 @@
+# -------------------------------------------------------
+# Add and Search Word - Data structure design - https://leetcode.com/explore/challenge/card/august-leetcoding-challenge/549/week-1-august-1st-august-7th/3413/
+# -------------------------------------------------------
+# Author: Arshad Mehmood
+# Github: https://github.com/arshad115
+# Blog: https://arshadmehmood.com
+# LinkedIn: https://www.linkedin.com/in/arshadmehmood115
+# Date : 2020-08-5
+# Project: leetcode-august-2020
+# -------------------------------------------------------
+import collections
+
+
+class WordDictionary:
+
+    def __init__(self):
+        """
+        Initialize your data structure here.
+        """
+        self.dict = collections.defaultdict(list)
+
+    def addWord(self, word: str) -> None:
+        """
+        Adds a word into the data structure.
+        """
+        if word:
+            self.dict[len(word)].append(word)
+
+    def search(self, word: str) -> bool:
+        """
+        Returns if the word is in the data structure. A word could contain the dot character '.' to represent any one letter.
+        """
+        if not word:
+            return False
+        if '.' not in word:
+            return word in self.dict[len(word)]
+        else:
+            for w in self.dict[len(word)]:
+                for i, l in enumerate(word):
+                    if l!=w[i] and l!='.':
+                        break
+                else:
+                    return True
+        return False
+
+# Your WordDictionary object will be instantiated and called as such:
+obj = WordDictionary()
+obj.addWord("bad")
+obj.addWord("dad")
+obj.addWord("mad")
+print(obj.search("pad"))
+print(obj.search("bad"))
+print(obj.search(".ad"))
+print(obj.search("b.."))
+

README.md

@@ -10,3 +10,4 @@ My leetcode profile can be viewed [here](https://leetcode.com/arshad115/)
 | 2    | [Design Hashset](./codes/2-design-hashset.py) |
 | 3    | [Valid Palindrome](./codes/3-valid-palindrome.py) |
 | 4    | [Power of Four](./codes/4-valid-palindrome.py) |
+| 4    | [Add and Search Word - Data structure design](./codes/5-add-and-search-word-data-structure-design.py) |