LeetCode 140. 单词拆分 II (动态规划)

Description

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，在字符串中增加空格来构建一个句子，使得句子中所有的单词都在词典中。返回所有这些可能的句子。

说明：

分隔时可以重复使用字典中的单词。 你可以假设字典中没有重复的单词。

示例 1： 输入: s = "catsanddog" wordDict = ["cat", "cats", "and", "sand", "dog"] 输出: [ "cats and dog", "cat sand dog" ] 示例 2： 输入: s = "pineapplepenapple" wordDict = ["apple", "pen", "applepen", "pine", "pineapple"] 输出: [ "pine apple pen apple", "pineapple pen apple", "pine applepen apple" ] 解释: 注意你可以重复使用字典中的单词。 示例 3： 输入: s = "catsandog" wordDict = ["cats", "dog", "sand", "and", "cat"] 输出: [] 来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/word-break-ii 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution

解法是在139. 单词拆分增加了状态的跟踪，时间复杂度$O(N^3)$，空间复杂度太高无法通过。

class Solution: def wordBreak(self, s: str, wordDict: List[str]) -> List[str]: n = len(s) dp = [False]*(n+1) states = [[] for _ in range(len(s)+1)] states[0].append('') dp[0] = True for i in range(n): for j in range(i, n): if(dp[i] and (s[i:j+1] in wordDict)): dp[j+1] = True for mid_str in states[i]: states[j+1].append(mid_str + ' ' + s[i:j+1]) # print(states) ans = states[-1] for i in range(len(ans)): ans[i] = ans[i].strip() return ans