[leetcode]Top K Frequent Elements

Belle唯唯

于 2019-08-28 22:31:43 发布

阅读量128

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/weixin_36869329/article/details/100127975

leetcode 专栏收录该内容

211 篇文章

订阅专栏

本文介绍了一种利用最大堆数据结构解决LeetCode上编号为347的题目——Top K频繁元素的方法。通过Java实现，首先使用HashMap统计每个元素出现的频率，然后创建一个最大堆对元素按频率进行排序，最后从堆中取出前K个最频繁的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用最大堆

import java.util.HashMap;
import java.util.Map;
import java.util.PriorityQueue;

/*
 * @lc app=leetcode id=347 lang=java
 *
 * [347] Top K Frequent Elements
 */
class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        if(nums == null || nums.length == 0) return new ArrayList<Integer>();
        HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
        List<Integer> list = new ArrayList<Integer>();
        for(int i : nums){
            map.put(i, map.getOrDefault(i, 0) + 1);
        }
        PriorityQueue<Map.Entry<Integer,Integer>> maxHeap = 
        new PriorityQueue<>((a, b) -> b.getValue() - a.getValue());
        for(Map.Entry<Integer, Integer> entry: map.entrySet()){
            maxHeap.add(entry);
        }

        while(list.size() < k){
            Map.Entry<Integer, Integer> entry = maxHeap.poll();
            list.add(entry.getKey());
        }
        
        return list;
        
    }
}