- 我们如何从一个链表或者数组中,找到第k大的数,或者前k大的数。使用小顶堆。(输出是从小到大排列的前K大的数)
- 我们如何从一个链表或者数组中,找到第k小的数,或者前k小的数(输出的堆是堆顶最大,其他不是序列顺序),使用大顶堆。
小顶堆找TopN元素:
可运行完整代码:
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) { // 选出前N大的数
if (queue.size() < N) { // 先比较,再添加(size=3,添加元素2,之后此if走完size就会变为4)
queue.offer(val);
} else if (queue.size() >= N && queue.peek() < val) { // 小顶堆堆顶存放堆中最小的数,如果有val比堆顶元素大,就d出最小(堆顶) 并 添加新元素
queue.poll();
queue.offer(val);
}
System.out.println("数组中第" + N + "大的数是:" + queue.peek());
System.out.println("从小到大前N个数的堆为:" + queue);
}
}
如何从大量数据中找出高频词?
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)