代码随想录算法训练营第四十六天｜139.单词拆分

139.单词拆分

https://leetcode.com/problems/word-break/description/

思路：判断一个字符串是否能由列表中的子字符串组成， 列表中的元素可以复用， 那么可以把这类问题转化成完全背包问题。 用 dp 数组的i+1记录到字符串的下标 i 是是否可以组成。 dp[0] 要初始化成 1. 然后考虑到这个问题是排列问题， 例如 字符串'leetcode' 列表['code', 'leet']. 所以应该先循环背包容量再循环数组。 如果 str[i-len(substr) :i] == substr, 并且 dp[i-len(substr)] 都是true 的时候， dp[i] = True. 或者 如果原本 dp[i] = True了， 就不要再往下循环数组了 （表示已经fit了）。

难点：递归的逻辑比较难想， 要考虑原本dp[i] = True 的情况

class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        dp = [0 for _ in range(len(s)+1)]
        dp[0] = 1
        words = [(word, len(word)) for word in wordDict]
        for i in range(1, len(dp)):
            for word, l in words:
                if i < l: continue
                dp[i] = int(word == s[i-l:i]) * dp[i-l]
                if dp[i] == 1: break
        return bool(dp[-1])