目录

0139:单词拆分(★)

力扣第 139 题

题目

给你一个字符串 s 和一个字符串列表 wordDict 作为字典。如果可以利用字典中出现的一个或多个单词拼接出 s 则返回 true

注意:不要求字典中出现的单词全部都使用,并且字典中的单词可以重复使用。

示例 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

提示:

  • 1 <= s.length <= 300
  • 1 <= wordDict.length <= 1000
  • 1 <= wordDict[i].length <= 20
  • swordDict[i] 仅由小写英文字母组成
  • wordDict 中的所有字符串 互不相同

相似问题:

分析

  • 按最后一个单词长度递推即可
  • 注意单词长度最多 20,可以剪枝

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        n = len(s)
        f = [1]+[0]*n
        vis = set(wordDict)
        for i in range(1,n+1):            
            for j in range(max(0,i-20),i):
                if s[j:i] in vis:
                    f[i] |= f[j]
        return f[-1]>0

31 ms

*附加

也可以用字典树加速查找。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        T = lambda: defaultdict(T)
        trie = T()
        for w in wordDict:
            p = trie
            for c in w[::-1]:
                p = p[c]
            p['#'] = ''
        n = len(s)
        f = [1]+[0]*n
        for i in range(1,n+1):         
            p = trie
            for j in range(i-1,max(0,i-20)-1,-1):
                if s[j] not in p:
                    break   
                p = p[s[j]]
                if '#' in p:
                    f[i] |= f[j]
        return f[-1]>0

47 ms