小白学算法1.4——优先队列
标签: 小白学算法
1.什么是优先队列
队列是先进先出,就像排队买东西一样,先排的人先买,后排的人后买,常见的操作是进队和出队。优先队列也像队列一样,常见的操作也是进队和出队,不过出队操作并不是把最早进入的元素出队,而是把最大或者最小的元素出队,这就是“优先队列”中“优先”的含义。
优先队列的出队和入队操作常常被称为:
- 插入元素(insert)
- 删除最大/最小元素(delMax/delMin)
2.实现优先队列的常见方法
优先队列的实现方法有多种,如下所示:
- 数组无序实现
- 可以通过数组实现的下压栈来实现优先队列。
insert
操作和push
操作一样,delMax操作中需要先遍历数组,找出最大的元素,然后将最大元素和栈顶元素交换位置,最后出栈
- 可以通过数组实现的下压栈来实现优先队列。
- 数组有序实现
- 可以通过数组实现的下压栈来实现优先队列。不过有序实现中需要修改的是insert操作,insert操作类似于插入排序,每次插入元素之后整个数组依旧有序,栈顶到栈底的元素按照从大到小的顺序排列。delMax操作和pop操作一样
- 链表无序实现,逻辑上和数组无序类似
- 链表有序实现,逻辑上和数组有序类似
- 二叉堆实现
优先队列不同的实现方法性能不同,优先队列各种实现最坏情况下运行时间的增长数量级如下:
数据结构 | 插入元素 | 删除元素 |
---|---|---|
无序数组 | 1 | N |
有序数组 | N | 1 |
二叉堆 | logN | logN |
很明显,二叉堆实现的优先队列性能最优,所以本文将使用二叉堆实现优先队列。
3.二叉堆实现优先队列
二叉堆的相关内容请阅读小白学算法1.3——堆。
3.1插入元素
首先将插入的元素放置在数组末尾,然后调用swim函数将其上浮到合适的位置。
void insert(int* a, int n, int k)
{
a[++n] = k;
swim(a, n);
}
3.2删除最大元素
首先将根节点删除,然后把数组的最后一个元素放到根节点位置,再调用sink函数将其下沉到合适的位置。
int delMax(int* a, int n, int k)
{
int max = a[1];
swap(a, 1, n--);
sink(a, n, 1);
return max;
}
注:n指向数组的最后一个元素,但并不表示数组的大小为n+1,数组大小应该用另外的一个变量来控制,比如用N表示。数组的大小根据n和N的关系动态分配。
4.总结
- 使用二叉堆实现优先队列比使用数组和链表实现优先队列高效许多
- 优先队列在选取多个最值的情况下很有用,比如10亿个数据选取100个最大值,使用优先队列很容易就可以办到,全部排序再选前100个数就很不可取
- 基于优先队列可以实现堆排序