[和小菜鸡一起刷题(python)] LeetCode 140. 单词拆分 II （Word Break II）

原题

给定一个非空字符串 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 139. 单词拆分的升级版，我们要给出各种拆分的方法。通常这种求所有可能答案的题目都会用到回溯算法，回溯算法是对解空间的深度优先搜索。例如输入示例一“catsanddog”，我们可以先将s分割成部分解“cat”和剩余字符串“sanddog”。利用分治的思想，递归地对剩余字符串进行切分。在代码中我先统计出wordDict中的单词长度来减少迭代次数。但是如果仅仅像这样做还是会超出时间限制。
对此采取的办法如下，我们在分割一个字符串之前先判断其能否被拆分，即把LeetCode 139照搬过来。若能拆分再进行上面的操作。

代码

class Solution(object):
    def wordBreak(self, s, wordDict):
        """
        :type s: str
        :type wordDict: List[str]
        :rtype: List[str]
        """
        res = []
        if len(s) == 0 or not wordDict:
            return []
        strides = [len(x) for x in wordDict]
        max_stride = max(strides)
        stride_set = set(strides)
        if self.canBreak(s, wordDict):
            self.sub_wordBreak(s, wordDict, max_stride, stride_set, [], res)
        return res
    
    def sub_wordBreak(self, sub_s, wordDict, max_stride, stride, cur_result, result):
        if len(sub_s) == 0:
            result.append(" ".join(cur_result))
        else:
            for stride_length in stride:
                if stride_length <= len(sub_s):
                    if sub_s[:stride_length] in wordDict:
                        self.sub_wordBreak(sub_s[stride_length:], wordDict, max_stride,stride, cur_result + [sub_s[:stride_length]], result)
            
    def canBreak(self, s, wordDict):
        can_make = [False] * (len(s)+1)         
        can_make[0] = True
        for i in range(1, len(s)+1):            
            for j in range(i-1, -1, -1):        
                if can_make[j] and s[j:i] in wordDict:
                    can_make[i] = True
                    break
        return can_make[-1]