单词拆分-leetcode

题目描述

给你一个字符串 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 中的所有字符串 互不相同

解法一

思路：

假设我们现在要计算 dp[i]（判断前 i 个字符是否合法）。 我们可以拿一个指针 j，在 0 到 i-1 之间划一刀，把这前 i 个字符切成两半：

1. 前半部分：前 j 个字符。看 dp[j]是否合法。
2. 后半部分：第 j 到 i-1 个字符。查看字串 s.substring(j, i) 在不在字典里。

只要能找到任意一个切分点 j，满足： dp[j] == true 并且 wordDict.contains(s.substring(j, i)) 那就说明前 i 个字符可以被字典里字符切分。于是立刻设 dp[i] = true，并 break。

代码：

public class Solution {
    public boolean wordBreak(String s, List<String> wordDict) {
        // 1. 【性能优化】把 List 转为 HashSet。查找单词时间从 O(N) 变成 O(1)。
        Set<String> wordDictSet = new HashSet(wordDict);
        
        // 2. dp 数组长度比字符串多 1，为了存下 dp[0] 和 dp[s.length()]
        boolean[] dp = new boolean[s.length() + 1];
        
        // 3. 边界条件：空串是合法的起点
        dp[0] = true;
        
        // 4. 外层循环：i 表示当前考察“前 i 个字符组成的子串”
        for (int i = 1; i <= s.length(); i++) {
            
            // 5. 内层循环：j 是切分点，把前 i 个字符切成 [0 到 j-1] 和 [j 到 i-1] 两半
            for (int j = 0; j < i; j++) {
                
                // 6. 核心判断：
                // dp[j] == true -> 前半段合法？
                // s.substring(j, i) -> 这就是你刚才问的：截取从 j 到 i-1 的字符构成的后半段
                // wordDictSet.contains(...) -> 后半段在字典里吗？
                if (dp[j] && wordDictSet.contains(s.substring(j, i))) {
                    dp[i] = true; // 整个前 i 个字符组合成功！
                    break;        // 只要找到一种正确切法，就不用再试其他 j 了，直接跳出内层循环
                }
            }
        }
        
        // 7. 返回最终结果：整个字符串（前 s.length() 个字符）能否拆分成功
        return dp[s.length()];
    }
}