ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

leetcode692_theKFrequentWord

2022-02-08 23:34:59  阅读:140  来源: 互联网

标签: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. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有