常用方法
1.add(e)
2.element()、peek()
element()
和peek()
的语义完全相同,都是获取但不删除队首元素,也就是队列中权值最小的那个元素,二者唯一的区别是当方法失败时前者抛出异常,后者返回null
。根据小顶堆的性质,堆顶那个元素就是全局最小的那个;由于堆用数组表示,根据下标关系,0
下标处的那个元素既是堆顶元素。所以直接返回数组0
下标处的那个元素即可。
3.remove()、poll()
remove()
和poll()
方法的语义也完全相同,都是获取并删除队首元素,区别是当方法失败时前者抛出异常,后者返回null
。由于删除操作会改变队列的结构,为维护小顶堆的性质,需要进行必要的调整。
示例代码
import java.util.*;
class Node
{
public int key;
public int sec;
}
public class Main
{
static Node []A = null;
public static void main(String args[])
{
Scanner in = new Scanner(System.in);
Comparator<Node> cmp = new Comparator<Node>() {
public int compare(Node a, Node b) { //这里是小根堆
if (a.key == b.key) return a.sec - b.sec; //第二键值做参考
return a.key - b.key; //第一键值做参考
}
};
Queue<Node> pq = new PriorityQueue<>(cmp);
int n = in.nextInt();
A = new Node [n];
for (int i = 0; i < n; i++) {
A[i] = new Node();
A[i].key = in.nextInt();
A[i].sec = in.nextInt();
pq.add(A[i]);
}
while (!pq.isEmpty()) {
System.out.println(pq.element().key + " " + pq.element().sec);
pq.poll();
}
}
}