题目描述
给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict,判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。
说明:
拆分时可以重复使用字典中的单词。
你可以假设字典中没有重复的单词。
示例 1:
输入: s = "leetcode", wordDict = ["leet", "code"]
输出: true
解释: 返回 true 因为 "leetcode" 可以被拆分成 "leet code"。
示例 2:
输入: s = "applepenapple", wordDict = ["apple", "pen"]
输出: true
解释: 返回 true 因为 "applepenapple" 可以被拆分成 "apple pen apple"。
注意你可以重复使用字典中的单词。
示例 3:
输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
输出: false
解题思路
动态规划
dp[i]表示以第i个下标结尾的字符串是否能有效拆分,对于第i个下标需要求解dp[i],需要枚举小于i的所有下标j,判断dp[j]和s[i:j]是否在字典中。
dp[i] = dp[j] and check(s[i: j]), 0 <= j < i
class Solution(object):
def wordBreak(self, s, wordDict):
m = len(s)
dp = [False] * (m + 1)
dp[0] = True
for i in range(1, len(dp)):
for j in range(i):
if dp[j] and s[j: i] in wordDict:
dp[i] = True
break
return dp[-1]
S = Solution()
print(S.wordBreak("catsandog", ["cats", "dog", "sand", "and", "cat"]))
记忆化递归
从下标1开始遍历每个下标i,如果s[0: i]在字典里,就继续递归。
class Solution(object):
def __init__(self):
self._dict = {}
def wordBreak(self, s, wordDict):
m = len(s)
def dfs(s):
if not s:
return True
if s in self._dict:
return self._dict[s]
res = False
for i in range(1, len(s) + 1):
if s[: i] in wordDict:
res = res or dfs(s[i: ])
self._dict[s] = res
return self._dict[s]
return dfs(s)
S = Solution()
print(S.wordBreak("abcd", ["a","abc","b","cd"]))