如何使用大顶堆和小顶堆找topN
- 我们如何从一个链表或者数组中,找到第k大的数,或者前k大的数。使用小顶堆。(输出是从小到大排列的前K大的数)
- 我们如何从一个链表或者数组中,找到第k小的数,或者前k小的数(输出的堆是堆顶最大,其他不是序列顺序),使用大顶堆。
小顶堆找TopN元素:
一、使用java.util.PriorityQueue优先级队列,默认是从小到大排序,初始指定最大容量为N
二、当size()大小<N的时候直接加到队列会自动排序,当size()>=N的时候将新val与堆顶最小的比较替换(其中queue.poll是弹出数组第一个元素,也即是堆顶的最小元素元素)
可运行完整代码:
package com.排序;
import java.util.PriorityQueue;
public class 小顶堆找最大TopN {
static int N = 4;
static PriorityQueue<Integer> queue;
public static void main(String[] args) {
queue = new PriorityQueue<Integer>(N);
int[] arr = {1, 3, 5, 2, 6};
for (int num : arr) {
小顶堆找TopN.add(num);
}
}
private static void add(int val) {
if (queue.size() < N) {
queue.offer(val);
} else if (queue.size() >= N && queue.peek() < val) {
queue.poll();
queue.offer(val);
}
System.out.println("数组中第" + N + "大的数是:" + queue.peek());
System.out.println("从小到大前N个数的堆为:" + queue);
}
}
如何从大量数据中找出高频词?
|