package utils;
import java.util.Collection;
import java.util.PriorityQueue;
public class TopN<E> {
private PriorityQueue<E> p; // 使用最小堆实现,根节点是最小的节点
private int k;
public TopN(int k) {
this.k = k;
this.p = new PriorityQueue<>(k);
}
/**
* 添加一个列表
*/
public void addAll(Collection<? extends E> collection) {
for (E e : collection) {
add(e);
}
}
/**
* 添加单个元素
*/
public void add(E e) {
if (p.size() < k) {
p.add(e);
}
Comparable<? super E> head = (Comparable<? super E>) p.peek(); // 查看头部元素,即最小的元素
if (head.compareTo(e) > 0) {
// head 小于 e ,不用改变
return;
}
// 新元素小于原来最小的元素,替换原来最小的元素成为topN之一
p.poll(); // 删除头部元素
p.add(e);
}
/**
* 返回当前最大的K个元素
*
* @param a T类型的数组
* @param <T> 声明此方法持有一个类型T,标识该方法为泛型方法“”
* @return T[] 返回T类型的数组
*/
public <T> T[] toArray(T[] a) {
return p.toArray(a); // 可以简单理解为:将所有元素复制到a中。
}
/**
* 获取第N个最大的元素
*
* @return
*/
public E getNth() {
return p.peek();
}
}
PriorityQueue实现:求前N个最大的元素
最新推荐文章于 2023-05-31 15:15:32 发布