Create word_pattern.py

yingl · web-flow · commit 03a0d0c1080a · 2019-04-25T16:09:08.000+08:00
diff --git a/word_pattern.py b/word_pattern.py
@@ -0,0 +1,28 @@
+class Solution:
+    """
+    @param pattern: a string, denote pattern string
+    @param teststr: a string, denote matching string
+    @return: an boolean, denote whether the pattern string and the matching string match or not
+    """
+    def wordPattern(self, pattern, teststr):
+        # write your code here
+        di = {}
+        pattern_mapping = []
+        _id = 0
+        for c in pattern:
+            if c not in di:
+                di[c] = _id
+                _id += 1
+            pattern_mapping.append(di[c])
+        _id = 0
+        di.clear()
+        str_mapping = []
+        words = teststr.split(' ')
+        for word in words:
+            if word not in di:
+                di[word] = _id
+                _id += 1
+            str_mapping.append(di[word])
+        return str_mapping == pattern_mapping
+
+# easy: https://www.lintcode.com/problem/word-pattern/
