Medium | LeetCode 139. 单词拆分 | 动态规划

139. 单词拆分

给定一个非空字符串 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

解题思路

动态规划dp[i] 表示前i个字符, 能够被空格拆分成词典中的词。在每次判断dp[i]时, 枚举[0,i)的每个j, 如果dp[j] = true, 并且(j, i)的字符串在字典中, 那么当前字符及其之前的字符能够拆分成字典中的词。

public boolean wordBreak(String s, List<String> wordDict) {
    Set<String> wordDictSet = new HashSet(wordDict);
    boolean[] dp = new boolean[s.length() + 1];
    dp[0] = true;
    for (int i = 1; i <= s.length(); i++) {
        // 计算dp[i]
        for (int j = 0; j < i; j++) {
            // 枚举[o,i)之间的每个j, 如果dp[j] == true, 并且[j,i)之间的字符串也在字典中
            if (dp[j] && wordDictSet.contains(s.substring(j, i))) {
                // 那么说明当前及其之前的字符, 能够拆分成字典中的数
                dp[i] = true;
                break;
            }
        }
    }
    return dp[s.length()];
}