PriorityQueue<Integer> queue = new PriorityQueue<Integer>(new Comparator<Integer>() { public int compare(Integer num1, Integer num2) { return num2 - num1; } }); 为什么以上代码是创建了一个大顶堆
时间: 2024-02-15 09:22:07 浏览: 75
以上代码创建了一个大顶堆的优先队列,这是因为在比较器的compare方法中,我们将第二个元素减去第一个元素,即num2 - num1。在大顶堆中,较大的元素具有更高的优先级,因此当第二个元素比第一个元素更大时,返回的结果是正数,表示第二个元素具有更高的优先级。这样,在插入元素时,较大的元素会被放在队列的前面,形成一个大顶堆的结构。
相关问题
PriorityQueue<Integer> queue = new PriorityQueue<Integer>(new Comparator<Integer>() { public int compare(Integer num1, Integer num2) { return num2 - num1; } }); 以上代码是创建了一个大顶堆还是小根堆
以上代码创建了一个大顶堆。在比较器的compare方法中,我们将第二个元素减去第一个元素,即num2 - num1。在大顶堆中,较大的元素具有更高的优先级,因此当第二个元素比第一个元素更大时,返回的结果是正数,表示第二个元素具有更高的优先级。这样,在插入元素时,较大的元素会被放在队列的前面,形成一个大顶堆的结构。
public int[] topKFrequent(int[] nums, int k) { Map<Integer, Integer> occurrences = new HashMap<Integer, Integer>(); for (int num : nums) { occurrences.put(num, occurrences.getOrDefault(num, 0) + 1); } // int[] 的第一个元素代表数组的值,第二个元素代表了该值出现的次数 PriorityQueue<int[]> queue = new PriorityQueue<int[]>(new Comparator<int[]>() { public int compare(int[] m, int[] n) { return m[1] - n[1]; } }); for (Map.Entry<Integer, Integer> entry : occurrences.entrySet()) { int num = entry.getKey(), count = entry.getValue(); if (queue.size() == k) { if (queue.peek()[1] < count) { queue.poll(); queue.offer(new int[]{num, count}); } } else { queue.offer(new int[]{num, count}); } } int[] ret = new int[k]; for (int i = 0; i < k; ++i) { ret[i] = queue.poll()[0]; } return ret; } 解析一下以上java代码
这段代码是一个用于获取数组中前k个出现频率最高的元素的函数。它使用了哈希表(HashMap)来记录每个元素出现的次数。然后,通过优先队列(PriorityQueue)来维护当前出现频率最高的k个元素。
代码首先遍历数组,使用哈希表occurrences记录每个元素出现的次数。接下来,创建一个优先队列queue,并通过传入一个自定义的比较器来确定元素之间的优先级。比较器中,通过比较数组中索引为1的元素来确定元素的优先级。
然后,遍历occurrences的每个键值对,将键和值构造成一个int数组,并将其插入到优先队列queue中。在插入过程中,如果队列的大小已经达到了k个,并且当前元素的出现次数比队列中最小优先级的元素的次数更高,则将最小优先级元素移除,再将当前元素插入队列。
最后,从优先队列queue中依次弹出前k个元素的第一个值(即数组中的值),并将它们存储在结果数组ret中。最后,返回结果数组ret作为函数的输出。
综上所述,该函数可以通过优先队列和哈希表来快速获取数组中前k个出现频率最高的元素。
阅读全文