class Solution {
public String longestWord(String[] words) {
Arrays.sort(words, (a, b) -> {
if (a.length() != b.length()) {
return a.length() - b.length();
} else {
return b.compareTo(a);
}
});
String longest = "";
Set<String> candidates = new HashSet<String>();
candidates.add("");
int n = words.length;
for (int i = 0; i < n; i++) {
String word = words[i];
if (candidates.contains(word.substring(0, word.length() - 1))) {
candidates.add(word);
longest = word;
}
}
return longest;
}
}
// 排序+哈希集合:
// 思路和算法
// 定义「符合要求的单词」如下:
// 空字符串是符合要求的单词;
// 在符合要求的单词的末尾添加一个字母,得到的新单词是符合要求的单词。
// 这道题要求返回数组words 中的最长的符合要求的单词,如果有多个最长的符合要求的单词则返回其中字典序最小的单词。以下将返回值称为「答案」
// 为了方便处理,需要将数组 words 排序,排序的规则是首先按照单词的长度升序排序,如果单词的长度相同则按照字典序降序排序。排序之后,可以确保当遍历到每个单词时,比该单词短的全部单词都已经遍历过,且每次遇到符合要求的单词一定是最长且字典序最小的单词,可以直接更新答案。
// 将答案初始化为空字符串。使用哈希集合存储所有符合要求的单词,初始时将空字符串加入哈希集合。遍历数组 words,对于每个单词,判断当前单词去掉最后一个字母之后的前缀是否在哈希集合中,如果该前缀在哈希集合中则当前单词是符合要求的单词,将当前单词加入哈希集合,并将答案更新为当前单词。
// 遍历结束之后,返回答案。