《LeetCode力扣练习》代码随想录——栈与队列（前 K 个高频元素---Java）

XRT_knives

于 2023-12-31 19:33:16 发布

阅读量415

点赞数 9

分类专栏： # LeetCode 文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/XRT_knives/article/details/135318857

版权

LeetCode 专栏收录该内容

181 篇文章 3 订阅

订阅专栏

本文介绍了如何使用Java实现LeetCode题目347——前K个高频元素，利用堆数据结构优化，统计输入数组中每个元素的频率并找出前k个最常见的元素。

摘要由CSDN通过智能技术生成

                    
                    《LeetCode力扣练习》代码随想录——栈与队列（前 K 个高频元素—Java） 
 
 刷题思路来源于 代码随想录 
 
347. 前 K 个高频元素 
 堆 class Solution {
    public int[] topKFrequent(int[] nums, int k) {

        if(nums.length==1){
            return nums;
        }

        Map<Integer,Integer> map=new HashMap<>();

        for(int num:nums){
            map.put(num,(map.getOrDefault(num,0)+1));
        }

        PriorityQueue<int[]> heap=new PriorityQueue<>((o1,o2)->o1[1]-o2[1]);

        for(Map.Entry<Integer,Integer> entry:map.entrySet()){

            if(heap.size()<k){
                heap.offer(new int[]{entry.getKey(),entry.getValue()});
            }else{

                if(entry.getValue()>heap.peek()[1]){
                    heap.poll();
                    heap.offer(new int[]{entry.getKey(),entry.getValue()});
                }

            }

        }

        int[] result=new int[k];

        for(int i=k-1;i>=0;i--){
            result[i]=heap.poll()[0];
        }

        return result;

    }
}