PriorityQueue实现：求前N个最大的元素

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();
    }
}