目录
连接词
描述
给你一个 不含重复 单词的字符串数组 words ,请你找出并返回 words 中的所有 连接词 。
连接词 定义为:一个完全由给定数组中的至少两个较短单词组成的字符串。
示例 1
输入:words = ["cat","cats","catsdogcats","dog","dogcatsdog","hippopotamuses","rat","ratcatdogcat"] 输出:["catsdogcats","dogcatsdog","ratcatdogcat"] 解释:"catsdogcats" 由 "cats", "dog" 和 "cats" 组成; "dogcatsdog" 由 "dog", "cats" 和 "dog" 组成; "ratcatdogcat" 由 "rat", "cat", "dog" 和 "cat" 组成。
示例 2
输入:words = ["cat","dog","catdog"] 输出:["catdog"]
提示
- words[i] 仅由小写字母组成
方法:动态规划
题目要求的是所有连接词,而连接词是由两个以上的较短字符串组成,所以一个连接词肯定是可以由其他较短的字符串拼接而成。
- 我们使用一个Set记录所有的单词,主要是为了去重、空字符串这种特殊的case,对单词数组words进行遍历,肯定先要把自己移除掉,因为不能自己构成自己。
- 判断是否能由目前Set(以不包含word)中的元素构成当前word,如果可以,将word将入结果集,具体判断过程如下:
- 如果Set集合没有元素或者字符串就是空的,直接返回false;
- 构造DP数组dp,降低时间复杂度,dp[i]表示word从[0, i)之间的字符串是否可以分解成Set集合已经包含的较短的字符串,dp[n]就表示可以完全分解掉;
- 假设现在要计算word[0, i - 1]是否可以形成连接词,可以分解为两部分:word[0, j - 1]和word[j, i],i最终可以取到n,这也就对应着DP数组的两部分:word[0, j - 1]对应dp[j], word[j, i]需要判断下Set集合中是否包含word[j, i],如果包含则dp[i] = true;
- 返回dp[n];
- 最后再将当前单词word加入Set进行下一轮的判断
class Solution {
public List<String> findAllConcatenatedWordsInADict(String[] words) {
Set<String> set = new HashSet<>();
List<String> ret = new ArrayList<>();
for (String word : words) {
set.add(word);
}
for (String word : words) {
if ("".equals(word)) {
continue;
}
set.remove(word);
if (canBreak(word, set)) {
ret.add(word);
}
set.add(word);
}
return ret;
}
private boolean canBreak(String s, Set<String> set) {
int n = s.length();
if (set.size() == 0 || n == 0) {
return false;
}
boolean[] dp = new boolean[n + 1];
dp[0] = true;
for (int i = 1; i < n + 1; i++) {
for (int j = 0; j < i; j++) {
if (!dp[j]) {
continue;
}
if (set.contains(s.substring(j, i))) {
dp[i] = true;
break;
}
}
}
return dp[n];
}
}