【桶排序、堆排序】Leetcode --- NO.347 前K个高频元素（Java）

题目描述

题目分析

• 这道题就是找出数组内出现频率前几的数组
• 那么首先要计算数字出现的频率
• 然后对频率进行排序，但是排序时不可打乱数字和频率的对应关系，找出前几大的数字

解法分析

• 累计数字出现的频率，这个可以通过map来累计，将数字当作键，将频率当作值
• 因为最后要返回的是出现频率前几的数字数组，所以频率和数字不可以分开
• 可以使用桶排序，第几个桶中存储的数据就出现了几次
• 或者使用堆排序【官方题解】
• 这里涉及到 PriorityQueue 类的使用，参考《玩转数据结构之优先队列和堆》以及《PriorityQueue详解》
  

代码

• 使用桶排序

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
    	// 累计频率
        Map<Integer, Integer> feq = new HashMap<>();
        for(int num:nums){
            feq.put(num, feq.getOrDefault(num, 0)+1);
        }
        List<Integer>[] buckets = new ArrayList[nums.length + 1];
        // 第 i 个桶中存储的数据出现了 i 次 
        for(int key : feq.keySet()){
            int times = feq.get(key);
            if(buckets[times] == null){
                buckets[times] = new ArrayList<>();
            }
            buckets[times].add(key);
        }
        // 从后向前遍历桶，存入结果数组
        List<Integer> res = new ArrayList<>();
        int len = buckets.length - 1;
        for(int i = len;i >= 0 && res.size() < k; i--){
            if(buckets[i] == null){
                continue;
            }
            // 如果这个桶中的数字比结果列表剩余的位置多，那就取一部分即可，因为都是一样的频率
            // 否则就全部放入结果列表中
            if(buckets[i].size() <= k-res.size()){
                res.addAll(buckets[i]);
            } else {
                res.addAll(buckets[i].subList(0, k-res.size()));
            }
        }
        // 将列表转换成数组
        int[] re = new int[k];
        for(int i = 0;i<k;i++){
            re[i] = (Integer)res.toArray()[i];
        }
        return re;
    }
}

• 官方题解【注意：官方题解返回的是 List 但是实际题目要求返回的是 int[] 】

class Solution {
  public List<Integer> topKFrequent(int[] nums, int k) {
    // 累计频率
    HashMap<Integer, Integer> count = new HashMap();
    for (int n: nums) {
      count.put(n, count.getOrDefault(n, 0) + 1);
    }

    // 初始化堆，频率越小的元素越靠前
    PriorityQueue<Integer> heap =
            new PriorityQueue<Integer>((n1, n2) -> count.get(n1) - count.get(n2));

    // 保持前 k 大频率的数字在堆中
    for (int n: count.keySet()) {
      heap.add(n);
      if (heap.size() > k)
        heap.poll();
    }

    // 返回的结果列表
    List<Integer> top_k = new LinkedList();
    while (!heap.isEmpty())
      top_k.add(heap.poll());
    Collections.reverse(top_k);
    return top_k;
  }
}

作者：LeetCode
链接：https://leetcode-cn.com/problems/top-k-frequent-elements/solution/qian-k-ge-gao-pin-yuan-su-by-leetcode/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。