leetcode.com 139. Word Break

문제 링크: https://leetcode.com/problems/word-break/description/

Author: skysign

leetcode.com 139. Word Break v2/main.py

@@ -0,0 +1,23 @@
+from typing import List
+
+
+class Solution:
+    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
+        dp: List[bool] = [False for _ in range(len(s))]
+        l = len(s)
+        bgns: List[int] = [0]
+
+        while len(bgns) > 0:
+            bgn = bgns.pop(0)
+
+            for word in wordDict:
+                if s[bgn:].startswith(word):
+                    bgn_new = bgn + len(word)
+                    if bgn_new == l:
+                        return True
+                    else:
+                        if not bgn_new in bgns and dp[bgn_new] == False:
+                            bgns.append(bgn_new)
+                            dp[bgn_new] = True
+
+        return False

leetcode.com 139. Word Break v2/test_main.py

@@ -0,0 +1,15 @@
+from unittest import TestCase
+from main import Solution
+
+class TestSolution(TestCase):
+    def test1_word_break(self):
+        sol = Solution()
+        self.assertEqual(True, sol.wordBreak("leetcode", ["leet","code"]))
+
+    def test2_word_break(self):
+        sol = Solution()
+        self.assertEqual(True, sol.wordBreak("applepenapple", ["apple","pen"]))
+
+    def test3_word_break(self):
+        sol = Solution()
+        self.assertEqual(False, sol.wordBreak("catsandog", ["cats","dog","sand","and","cat"]))