题目描述:给一非空的单词列表,返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。
示例 1:
输入: [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2
输出: [“i”, “love”]
解析: “i” 和 “love” 为出现次数最多的两个单词,均为2次。
注意,按字母顺序 “i” 在 “love” 之前。
链接:https://leetcode-cn.com/problems/top-k-frequent-words
求解思路:
- 用 Map 将每个单词以及出现的次数统计;
- 将这些单词放入一个 ArrayList 中,然后进行排序;
- 在排序方法 sort() 的第二个参数中需传入一个 Comparator 对象,用来指定排序的比较规则(如果两单词出现的次数相等,就按照字符串的字典序来排序,即可使用CompareTo()方法);
- 最后使用方法 subList(0,k)取前 k 个高频单词。
代码实现:
class Solution {
public List<String> topKFrequent(String[] words, int k) {
//1.用Map统计words中每个单词出现的次数
Map<String,Integer> map = new HashMap<>();
for(String word : words){
Integer value = map.get(word);
if(value == null){
map.put(word,1);
}else{
map.put(word,value + 1);
}
}
//2.将所有的key取出来放到一个List中
List<String> wordList = new ArrayList<>(map.keySet());
//3.对 wordList 进行排序。
// 需在 sort 方法的第二个参数中传入一个Comparator对象,用来指定比较规则
Collections.sort(wordList, new Comparator<String>() { //java的匿名内部类
@Override
public int compare(String o1, String o2) {
Integer count1 = map.get(o1);
Integer count2 = map.get(o2);
if(count1.equals(count2)){//如果出现次数相等,就按字符串的字典序来排序
return o1.compareTo(o2);
}
return count2 - count1;
}
});
return wordList.subList(0,k);
}
}