diff --git a/src/my_project/interviews/top_150_questions_round_16/word_pattern.py b/src/my_project/interviews/top_150_questions_round_16/word_pattern.py new file mode 100644 index 00000000..92adb8f1 --- /dev/null +++ b/src/my_project/interviews/top_150_questions_round_16/word_pattern.py @@ -0,0 +1,23 @@ +from typing import List, Union, Collection, Mapping, Optional +from abc import ABC, abstractmethod +from collections import defaultdict + +class Solution: + def wordPattern(self, pattern: str, s: str) -> bool: + + dic_answer = defaultdict(str) + + words = s.split(' ') + + if len(words) != len(pattern) or len(set(words)) != len(set(pattern)): + return False + else: + for k, v in enumerate(words): + if v in dic_answer: + if dic_answer[v] != pattern[k]: + return False + else: + dic_answer[v] = pattern[k] + + return True + diff --git a/tests/test_150_questions_round_12/test_invert_binary_tree_round_12.py b/tests/test_150_questions_round_12/test_invert_binary_tree_round_12.py index bc945cd9..42ae4ee8 100644 --- a/tests/test_150_questions_round_12/test_invert_binary_tree_round_12.py +++ b/tests/test_150_questions_round_12/test_invert_binary_tree_round_12.py @@ -1,5 +1,5 @@ import unittest -from my_project.interviews.top_150_questions_round_12\ +from src.my_project.interviews.top_150_questions_round_12\ .invert_binary_tree import Solution, TreeNode class InvertTreeTestCase(unittest.TestCase): diff --git a/tests/test_150_questions_round_16/test_word_pattern_round_16.py b/tests/test_150_questions_round_16/test_word_pattern_round_16.py new file mode 100644 index 00000000..f251ce2e --- /dev/null +++ b/tests/test_150_questions_round_16/test_word_pattern_round_16.py @@ -0,0 +1,20 @@ +import unittest +from src.my_project.interviews.top_150_questions_round_16\ +.word_pattern import Solution + +class WordPatternTestCase(unittest.TestCase): + + def test_is_word_pattern(self): + solution = Solution() + output = solution.wordPattern(pattern="abba", s="dog cat cat dog") + self.assertTrue(output) + + def test_is_no_word_pattern_one(self): + solution = Solution() + output = solution.wordPattern(pattern="abc", s="dog cat cat fish") + self.assertFalse(output) + + def test_is_no_word_pattern_two(self): + solution = Solution() + output = solution.wordPattern(pattern="aa", s="dog cat cat fish") + self.assertFalse(output) \ No newline at end of file diff --git a/tests/test_150_questions_round_7/test_binary_tree_round_7.py b/tests/test_150_questions_round_7/test_binary_tree_round_7.py index d547b6cb..17cdc884 100644 --- a/tests/test_150_questions_round_7/test_binary_tree_round_7.py +++ b/tests/test_150_questions_round_7/test_binary_tree_round_7.py @@ -1,5 +1,5 @@ import unittest -from my_project.interviews.top_150_questions_round_7\ +from src.my_project.interviews.top_150_questions_round_7\ .invert_binary_tree import TreeNode, Solution class InvertTreeTestCase(unittest.TestCase): diff --git a/tests/test_150_questions_round_9/test_number_of_bits_round_9.py b/tests/test_150_questions_round_9/test_number_of_bits_round_9.py index 93737748..9843c2d1 100644 --- a/tests/test_150_questions_round_9/test_number_of_bits_round_9.py +++ b/tests/test_150_questions_round_9/test_number_of_bits_round_9.py @@ -1,5 +1,5 @@ import unittest -from my_project.interviews.top_150_questions_round_9\ +from src.my_project.interviews.top_150_questions_round_9\ .number_of_one_bits import Solution class NumberOfOnesTestCase(unittest.TestCase):