原题链接:https://leetcode-cn.com/problems/word-break-ii/。
这个题是139.单词拆分的拓展,要将所有的可能情况输出。
这里同样用动态规划的思想输出一个状态表,迭代式和单词拆分的一样。
然后利用dfs算法遍历,输出句子结果。这个思路与131.分割回文串思想完全相同。
下面贴python代码片。
class Solution(object):
def dfs(self,low,length,s,res,vaild,path=[]):
if low == length:
res.append(' '.join(path))
return
for i in range(low,length):
if vaild[low][i]:
self.dfs(i+1,length,s,res,vaild,path+[s[low:i+1]])
def wordBreak(self, s, wordDict):
"""
:type s: str
:type wordDict: List[str]
:rtype: List[str]
"""
length = len(s)
vaild = [[False]*length for _ in range(length)]
dp = [False for _ in range(length+1)]
dp[0] = True
wordDict = set(wordDict)
for i in range(1,length+1):
for j in range(i+1):
word = s[j:i]
if dp[j] and word in wordDict:
dp[i] = True
vaild[j][i-1] = True # 这一句表示的是s[j:i+1]是在单词字典中
res = []
if dp[-1]:
self.dfs(0,length,s,res,vaild)
return res