题目描述:
给一非空的单词列表,返回前 k 个出现次数最多的单词。
返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率,按字母顺序排序。
示例 :
输入: [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2
输出: [“i”, “love”]
解析: “i” 和 “love” 为出现次数最多的两个单词,均为2次。
注意,按字母顺序 “i” 在 “love” 之前。
解法1:哈希表存储,再通过集合容器排序。
class Solution {
public List<String> topKFrequent(String[] words, int k) {
//创建一个Map用于对word计数
Map<String,Integer> cnt = new HashMap<>();
for(String word:words){
cnt.put(word,cnt.getOrDefault(word,0)+1)
}
//创建一个list存放key
List<String> rec = new ArrayList<>();
//这里用到了entrySet方法获得了cnt的键值对集合
for(Map.Entry<String,Integer> entry : cnt.entrySet()){
rec.add(entry.getKey());
}
//利用collections方法排序
Collections.sort(rec, new Comparator<String>(){
public int compare(String word1,String word2){
return cnt.get(word1)==cnt.get(word2)?word1.compareTo(word2):cnt.get(word2)-cnt.get(word1);
}
});
return rec.subList(0,k);
}
解法2:利用PriorityQueue优先排序
class Solution {
public List<String> topKFrequent(String[] words, int k) {
//创建一个Map用于对word计数
Map<String,Integer> cnt = new HashMap<>();
for(String word:words){
cnt.put(word,cnt.getOrDefault(word,0)+1);
}
PriorityQueue<Map.Entry<String,Integer>> pq = new PriorityQueue<>(new Comparator<Map.Entry<String,Integer>>(){
public int compare(Map.Entry<String,Integer> entry1,Map.Entry<String,Integer> entry2){
return entry1.getValue()==entry2.getValue()?entry2.getKey().compareTo(entry1.getKey()):entry1.getValue()-entry2.getValue();
}
});
for(Map.Entry<String,Integer> entry:cnt.entrySet()){
pq.offer(entry);
if(pq.size()>k){
pq.poll;
}
}
List<String> ret = new ArrayList<>();
while(!pq.isEmpty()){
ret.add(pq.poll().getKey());
}
Collections.reverse(ret);
return ret;
}