PriorityQueue是一个优先队列,它实现了Queue接口,前面几篇文章我们分析过LinkedList与ArrayDeque的源码,它们也都实现了Queue接口,PriorityQueue与它们的不同之处在于PriorityQueue里面保存的元素有优先级的概念,每一个元素都有一个优先级,队头元素的优先级最高的,如果从队头元素依次出队列,得到的将是一个有序的输出。PriorityQueue的内部是用堆来实现的,堆是一棵完全二叉树,学过数据结构的同学应该都清楚二叉树结点的表示:
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
上面就是一棵二叉树的结点经典表示了,结点里面的left与right结点分别指向其左孩子与右孩子。但是由于堆是一棵完全二叉树,我们有更方便的方法求出一个结点的双亲结点与左右孩子结点,我们给一棵完全二叉树按层序,从左到右编号1,2,3...可得到如下图一张完全二叉树:
我们可以发现,如果一个结点的编号是 i,那么它的双亲结点的编号为 i/2 .左孩子结点为 2 * i ,右孩子结点为 2 * i + 1。如果将编号1,2,3...看做是数组的索引,那么我们就可以很方便的使用数组来求储存这棵完全二叉树了。
堆可以分为最小堆与最大堆,最小堆的性质是:每个结点都不小于其父节点;最大堆:每个结点都不大于其父节点。上面的例子是一个最小堆,首元素就是队首元素,优先级最高。
在堆中插入一个元素:
1、添加元素到最后的位置。
2、与父节点比较,如果小于等于父节点,则满足堆的性质,结束。否则与父节点进行交换,然后在与父节点比较交换,直到父节点为空或者大于等于父节点。
如果我们在队列中插入6,首先将6添加到队尾。
比较6(编号为10)与其父节点(编号为10/2=5)之间的大小,发现6比19小,与父节点交换。
比较6(现在编号为5)与其父节点(编号为5/2=2)之间的大小,发现6比2大,满足最小堆的性质,插入结束。
在堆中删除元素:
1、用最后一个结点替换待删除的元素,并删掉最后一个元素;
2、如果替换后的结点小于其父节点,则与父节点交换,直到满足最小堆的条件;如果替换后的结点大于某个孩子结点,则与较小的孩子结点交换,直到满足最小堆的条件。
如果我们删除上面最小堆中的3,首先用最后一个结点19,替换3,并删除19。