leetcode：单词拆分2（python）

1. 题目描述

给定一个非空字符串 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”]
输出:
[]

2. 思路

先判断能不能拆分，如果能拆分的话，先根据wordDict中的单词长度，从前往后分割，然后进行匹配，如果匹配上wordDict里的单词，对剩下的进行递归。直到len(s)==0,说明s都已经匹配完了。由于可能有不同的分割结果，每一个不同的分割结果只需要记录一次，因此需要对wordDict里的单词长度进行去重。

2.1 python 代码

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 []
        wordLenList = [len(word) for word in wordDict]
        
        wordLenSet = set(wordLenList)
        if self.judge(s,wordDict):
            self.dfs(s,wordDict,[],res,wordLenSet)
        return res
  
    def judge(self,s,wordDict):   # 先判断能不能分割
        lenth = len(s)
        dp = [False for i in range(lenth + 1)]
        dp[0] = True
        for i in range(1,lenth+1):
            for j in range(i):
                if dp[j] and s[j:i] in wordDict:
                    dp[i] = True
                    break
        return dp[-1]
    def dfs(self,s,wordDict,curResult,res,wordLenSet):
        if len(s) == 0:  # 递归终止条件 
            res.append(" ".join(curResult))
            return
        else:
            for i in wordLenSet:   # 进行回溯
                if i <= len(s):
                    if s[:i] in wordDict:
                        self.dfs(s[i:],wordDict,curResult+[s[:i]],res,wordLenSet)
            return