Top K Frequent Elements

最新推荐文章于 2024-08-18 16:33:37 发布

sjphiChina

最新推荐文章于 2024-08-18 16:33:37 发布

阅读量251

点赞数

CC 4.0 BY-SA版权

分类专栏： strStr 1 基本功候着常做常新

本文链接：https://blog.youkuaiyun.com/sjphiChina/article/details/52037947

strStr 同时被 3 个专栏收录

360 篇文章

订阅专栏

基本功

78 篇文章

订阅专栏

68 篇文章

订阅专栏

本文介绍了一种使用堆数据结构来找出数组中出现频率最高的Top K个元素的方法。通过构建哈希表记录每个元素的出现次数，并利用最小堆进行筛选，最终返回出现频率最高的K个元素。此方法适用于需要快速找到最频繁元素的应用场景。

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

先贴出用了heap的做法，稍后要学会用桶排序来做。

public class Solution {
    public List<Integer> topKFrequent(int[] nums, int k) {
        List<Integer> res = new LinkedList<>();
        if (nums == null || nums.length == 0) {
            return res;
        }
        Map<Integer, Integer> map = new HashMap<>();
        for (int i: nums) {
            if (map.containsKey(i)) {
                map.put(i, map.get(i) + 1);
            } else {
                map.put(i, 0);
            }
        }
        PriorityQueue<Pair> queue = new PriorityQueue<>(11, new Comparator<Pair>(){
            @Override
            public int compare(Pair a, Pair b) {
                return a.count - b.count;
            }
        });
        for (Map.Entry<Integer, Integer> entry: map.entrySet()) {
            queue.offer(new Pair(entry.getKey(), entry.getValue()));
            if (queue.size() > k) {
                queue.poll();
            }
        }
        while (!queue.isEmpty()) {
            res.add(0, queue.poll().num);
        }
        return res;
    }
    
    class Pair{
        int num;
        int count;
        public Pair (int n, int c) {
            num = n;
            count = c;
        }
    }
}