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

原题

给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。

说明：

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

示例 1：

输入: s = “leetcode”, wordDict = [“leet”, “code”]
输出: true
解释: 返回 true 因为 “leetcode” 可以被拆分成 “leet code”。

示例 2：

输入: s = “applepenapple”, wordDict = [“apple”, “pen”]
输出: true
解释: 返回 true 因为 “applepenapple” 可以被拆分成 “apple pen apple”。
注意你可以重复使用字典中的单词。

示例 3：

输入: s = “catsandog”, wordDict = [“cats”, “dog”, “sand”, “and”, “cat”]
输出: false

思路

此题用到的思想还是动态规划，而动态规划中我认为最最经典的例子还是爬楼梯问题。此题也可以类比为爬楼梯问题来方便理解。
例如输入为示例3：
|c 　|a　 |t　 |s　 |a　 |n　 |d　 |o　 |g　 |
0　１　２　３　４　５　６　７　８　９　
相当于判断能不能从位置0爬到位置9，只不过仍然是在能够前进的步数上有附加条件，即所跨出的步数内的字符串要出现在wordDict中。如能从0到3或4，因为cat和cats都是wordDict中的词。
有了以上的分析，我们可以建立一个长度为len(s)+1的list来记录我们是否能到达每一个位置，这里我用0和1表示。这个list的第一个元素为1，因为就是从这里起步的，最后返回最后一个元素的值作为答案。采用两层循环，第一层循环为查看每一个终止位置i；第二层循环为启示位置j(j<i)。如果list[j]为1说明我们能到达位置j，同时如果s[j:i]在wordDict中，说明我们可以从j跨到i，从而给list[i]标记为1。这里我加入了一个max_stride来减少j的迭代次数，因为如果我们最多一次跨5步，那我们并不需要关心5步之前位置的状态。

代码

class Solution(object):
    def wordBreak(self, s, wordDict):
        """
        :type s: str
        :type wordDict: List[str]
        :rtype: bool
        """
        if len(s) == 0 or not wordDict:
            return False
        max_stride = max([len(x) for x in wordDict])
        res = [0] * (len(s) + 1)
        res[0] = 1
        for i in range(1, len(s) + 1):
            for j in range(i-max_stride, i):
                if res[j] == 1 and s[j:i] in wordDict:
                    res[i] = 1
        if res[-1] == 1:
            return True
        else:
            return False