java单词拆分（Leetcode 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，其中 dp[i] 表示字符串 s 的前 i 个字符是否可以被拆分成字典中的单词。

状态转移方程

对于每一个位置 i，我们需要遍历所有可能的拆分位置 j（0 <= j < i），判断从位置 j 到位置 i 的子串是否在字典中。状态转移方程为：

dp[i] = dp[j] && check(s[j:i])

其中，check(s[j:i]) 表示判断字符串 s 的子串 s[j:i] 是否在字典中。

初始化

我们需要对动态规划数组进行初始化。初始时，dp[0] = true 表示空串可以被拆分。

Java解题

class Solution {
    public boolean wordBreak(String s, List<String> wordDict) {
        // 将字典中的单词存入哈希集合，便于快速查找
        Set<String> wordSet = new HashSet<>(wordDict);
        // 创建一个动态规划数组 dp，其中 dp[i] 表示字符串 s 的前 i 个字符是否可以被拆分成字典中的单词
        boolean[] dp = new boolean[s.length() + 1];
        // 初始化动态规划数组 dp
        dp[0] = true;

        // 遍历动态规划数组 dp
        for (int i = 1; i <= s.length(); i++) {
            // 遍历所有可能的拆分位置 j
            for (int j = 0; j < i; j++) {
                // 如果 dp[j] 为 true，且子串 s[j:i] 在字典中，则更新 dp[i] 为 true
                if (dp[j] && wordSet.contains(s.substring(j, i))) {
                    dp[i] = true;
                    break;
                }
            }
        }

        // 返回字符串 s 是否可以被拆分成字典中的单词
        return dp[s.length()];
    }
}

总结

通过动态规划的思想，我们可以解决这个问题。首先初始化动态规划数组，然后根据状态转移方程进行状态转移，最终判断字符串是否可以被拆分成字典中的单词。

优化彩蛋

一开始写的排名最后，就想这搞一下优化，发现其实这个优化真的很小很小。

第二次，加了一个break，可以减少多余的执行次数

第三次，再然后把一个临时变量删了，快了1ms（可能是积少成多，因为比较的次数多，每次都新建的时间堆起来就多了那么1ms吧哈哈哈）

第四次，想着 1、length（）方法是要计算的，就用临时变量保存一下，不用每次都调用一次（不像数组是对象属性可以直接.出来，是数组的话应该没有影响）2、把list改为set，hashset的查询效率复杂度为常量，能优化contains方法的时间。