题目描述:
给定一个不含重复单词的列表,编写一个程序,返回给定单词列表中所有的连接词。
连接词的定义为:一个字符串完全是由至少两个给定数组中的单词组成的。
示例:
输入: [“cat”,“cats”,“catsdogcats”,“dog”,“dogcatsdog”,“hippopotamuses”,“rat”,“ratcatdogcat”]
输出: [“catsdogcats”,“dogcatsdog”,“ratcatdogcat”]
解释: “catsdogcats"由"cats”, “dog” 和 "cats"组成;
“dogcatsdog"由"dog”, "cats"和"dog"组成;
“ratcatdogcat"由"rat”, “cat”, "dog"和"cat"组成。
说明:
给定数组的元素总数不超过 10000。
给定数组中元素的长度总和不超过 600000。
所有输入字符串只包含小写字母。
不需要考虑答案输出的顺序。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/concatenated-words
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
前缀树+回溯
代码:
注意引用数组的新建
class Solution {
class Trim{
boolean isleaf;
char word;
Trim[] next = new Trim[26]; //这里必须是new 具体的
String string;
}
public List<String> findAllConcatenatedWordsInADict(String[] words) {
// 建立前缀树
Trim trim = new Trim();
for (String string : words) {
Trim root = trim;
for (char string1 : string.toCharArray()) {
if(root.next[string1 - 'a'] == null){
root.next[string1 - 'a'] = new Trim();
root.next[string1 - 'a'].word = string1;
}
root = root.next[string1 - 'a'];
}
root.isleaf = true;
root.string = string;
}
List<String> result = new ArrayList<>();
for (String string : words) {
if(getcount(string, trim, 0)){
result.add(string);
}
}
return result;
}
public boolean getcount(String tem,Trim root,int count){
if((tem == null || tem.length() == 0 ) && count > 1){
return true;
}
Trim temTrim = root;
for (int i = 0; i < tem.length(); i++) {
char is = tem.charAt(i);
if(temTrim.next[is - 'a'] != null){
temTrim = temTrim.next[is - 'a'];
if(temTrim.isleaf){
if(getcount(tem.substring(i + 1), root, count + 1)){
return true;
}
}
}else{
return false;
}
}
return count > 1 && temTrim.isleaf;
}
}