与动态规划的爱恨情仇——单词拆分

95 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第15天,点击查看活动详情


最近一直在力扣刷题,也逐渐对各类题型有了自己的理解,所谓见招拆招,将自己的浅显经验分享一下,帮助更多在编程路上的朋友们。


单词拆分

给你一个字符串 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] 代表字典中能否组成从 0 到 i 位置的前缀。

初始化dp[0] = true,对于空字符串,不需要使用字典中的字符串即可组成。

状态转移方程如下:dp[i] = dp[j] && wordDict.contains(j, i),j 代表从 0 到 i - 1 的任一位置。

题解

class Solution {
    public boolean wordBreak(String s, List<String> wordDict) {
        int n = s.length();
        boolean[] dp = new boolean[n + 1];
        dp[0] = true;
        for(int i = 1; i <= n; i++) {
            for(int j = 0; j < i; j++) {
                dp[i] = wordDict.contains(s.substring(j, i)) && dp[j];
                if(dp[i]) {
                    break;
                }
            }
        }
        return dp[n];
    }
}

关于数据结构的优化

List集合底层是由数组实现的,而Set是由数组+链表+红黑树实现的。

故对于List集合来说,contains()方法并不高效,需要对其中的每一个元素进行遍历比较,而使用Set的contains()方法可以提高检索效率。

Set<String> set = new HashSet(wordDict);

可以将list集合中的元素存入Set中,再利用Set的contains()方法进行检索。