持续创作,加速成长!这是我参与「掘金日新计划 · 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()方法进行检索。