LeetCode--347.前k个高频元素(使用优先队列解决)

题目:给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

思路:先使用HashMap统计数组各个元素的个数,其中key代表数组元素,value代表元素出现的次数。然后使用队列进行map排序,排序是从大到小,选取前k个元素即为前k个高频元素

何为优先队列?

函数定义如下:

java 复制代码
public class PriorityQueue<E> extends AbstractQueue<E>
    implements java.io.Serializable {

JDK解释如下:

基于优先级堆的无限优先级队列。优先级队列的元素根据其自然 Sequences 进行排序,或者根据队列构造时提供的 进行排序Comparator(自定义排序规则),具体取决于使用的构造函数。优先级队列不允许null元素。依赖于自然排序的优先级队列也不允许插入不可比较的对象(这样做可能会导致 )。ClassCastException

代码及其解释:

java 复制代码
public int[] topKFrequent(int[] nums, int k) {
    // map中的key为数组元素,value为该元素出现的次数
    HashMap<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        if (map.containsKey(nums[i])) {
            map.put(nums[i], map.get(nums[i]) + 1);
        } else {
            map.put(nums[i], 1);
        }
    }
    // 优先队列,(o1, o2) -> map.get(o2) - map.get(o1) 升序排列
    PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> map.get(o2) - map.get(o1));
    // 遍历map,将元素放入优先队列中,会自动排序
    for (int i : map.keySet()) {
        queue.add(i);
    }
    int[] result = new int[k];
    for (int i = 0; i < k; i++) {
        result[i] = queue.poll();
    }
    return result;
}
相关推荐
风象南2 小时前
SpringBoot中6种自定义starter开发方法
java·spring boot·后端
mghio11 小时前
Dubbo 中的集群容错
java·微服务·dubbo
咖啡教室15 小时前
java日常开发笔记和开发问题记录
java
咖啡教室16 小时前
java练习项目记录笔记
java
鱼樱前端16 小时前
maven的基础安装和使用--mac/window版本
java·后端
RainbowSea17 小时前
6. RabbitMQ 死信队列的详细操作编写
java·消息队列·rabbitmq
RainbowSea17 小时前
5. RabbitMQ 消息队列中 Exchanges(交换机) 的详细说明
java·消息队列·rabbitmq
算AI18 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
李少兄18 小时前
Unirest:优雅的Java HTTP客户端库
java·开发语言·http