给定一个非空字符串 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
解法一:dfs
超时了,
class Solution:
def wordBreak(self, s: str, wordDict: List[str]) -> bool:
def dfs(state, flag):
if state in wordDict:
return True
for i in range(1, len(state)):
if state[:i] in wordDict:
flag = dfs(state[i:], flag)
if flag==True:
return True
return False
return dfs(s, False)
解法二 动态规划
dp[i]表示s的前 i 个字符能否拆成wordDict中的字符。
class Solution(object):
def wordBreak(self, s, wordDict):
"""
:type s: str
:type wordDict: List[str]
:rtype: bool
"""
dp = [False] * (len(s) + 1)
# dp[0]为True是为了能判断当j==0时s[0:i]是否在wordDict中
dp[0] = True
for i in range(1, len(s)+1):
for j in range(i):
# 当 前j个字符串为True时,判断s[j:i]是否也在wordDict中,若在,则为True,表示前i个字符可以全部拆成wordDict中的字符
if dp[j] and (s[j:i] in wordDict):
dp[i] = True
break
return dp[-1]
倒着判断似乎速度更快
class Solution(object):
def wordBreak(self, s, wordDict):
"""
:type s: str
:type wordDict: List[str]
:rtype: bool
"""
dp = [False] * (len(s) + 1)
dp[0] = True
for i in range(1, len(s)+1):
###############################
for j in range(i-1, -1, -1):
if dp[j] and (s[j:i] in wordDict):
dp[i] = True
break
return dp[-1]
优化:不用每次从s[0]开始搜索,因为wordDict中的字符串长度是有限的。只需要从i-max_len开始搜索就可以了。
class Solution(object):
def wordBreak(self, s, wordDict):
"""
:type s: str
:type wordDict: List[str]
:rtype: bool
"""
max_len = 0
for word in wordDict:
max_len = max(max_len, len(word))
dp = [False] * (len(s) + 1)
dp[0] = True
for i in range(1, len(s)+1):
for j in range(max(0, i-max_len), i):
if dp[j] and (s[j:i] in wordDict):
dp[i] = True
break
return dp[-1]