0139：单词拆分（★）

2015-05-19 约 432 字预计阅读 1 分钟次阅读

力扣第 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
s 和 wordDict[i] 仅由小写英文字母组成
wordDict 中的所有字符串 互不相同

分析

按第一个单词长度递归即可。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13


class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        @cache
        def dfs(s):
            if not s:
                return True
            for i in range(1,min(len(s),m)+1):
                if s[:i] in vis and dfs(s[i:]):
                    return True
            return False
        vis = set(wordDict)
        m = max(len(w) for w in vis)
        return dfs(s)

46 ms

*附加

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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22


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:
                p=p[c]
            p['#'] = ''
        @cache
        def dfs(i):
            if i==len(s):
                return True
            p = trie
            for j in range(i,len(s)):
                if s[j] not in p:
                    return False
                p = p[s[j]]
                if '#' in p and dfs(j+1):
                    return True
            return False
        return dfs(0)

37 ms

目录

0139：单词拆分（★）

题目

分析

解答

*附加