标签:Map stringIntegerEntry getValue theKFrequentWord Entry entry t1 leetcode692
class Solution {
public List<String> topKFrequent(String[] words, int k) {
Map<String, Integer> map = new HashMap<>();
for(String word: words) map.put(word, map.getOrDefault(word, 0) + 1);
PriorityQueue<Map.Entry<String, Integer>> priorityQueue = new PriorityQueue<>(new Comparator<Map.Entry<String, Integer>>() {
@Override
public int compare(Map.Entry<String, Integer> stringIntegerEntry, Map.Entry<String, Integer> t1) {
if(stringIntegerEntry.getValue() == t1.getValue()) {
// ascending
return t1.getKey().compareTo(stringIntegerEntry.getKey());
}
else {
// deascending
return stringIntegerEntry.getValue() - t1.getValue();
}
}
});
for(Map.Entry<String, Integer> entry: map.entrySet()) {
String key = entry.getKey();
int count = entry.getValue();
if(priorityQueue.size() == k) {
Map.Entry<String, Integer> top = priorityQueue.peek();
String topKey = top.getKey();
int topCount = top.getValue(); // aaa aa
if(count > topCount ||(count == topCount && key.compareTo(topKey) < 0)) {
priorityQueue.poll();
priorityQueue.offer(entry);
}
}
else priorityQueue.offer(entry);
}
List<Map.Entry<String, Integer>> list = new ArrayList<>();
for(Map.Entry<String, Integer> entry: priorityQueue) list.add(entry);
Collections.sort(list, new Comparator<Map.Entry<String, Integer>>() {
@Override
public int compare(Map.Entry<String, Integer> stringIntegerEntry, Map.Entry<String, Integer> t1) {
if(stringIntegerEntry.getValue() == t1.getValue()) {
// ascending
return stringIntegerEntry.getKey().compareTo(t1.getKey());
}
else {
// deascending
return t1.getValue() - stringIntegerEntry.getValue();
}
}
});
List<String> list1 = new ArrayList<>();
for(Map.Entry<String, Integer> entry: list) {
list1.add(entry.getKey());
}
return list1;
}
}
标签:Map,stringIntegerEntry,getValue,theKFrequentWord,Entry,entry,t1,leetcode692 来源: https://www.cnblogs.com/huangming-zzz/p/15873284.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。