LeetCode.140.单词拆分II

原题链接: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
                

你可能感兴趣的:(算法)