优先队列内部维持了一个堆,堆的特点是堆顶元素最大(或最小),利用优先队列查找最小的k个数的方法:
1、把前k个数当成一个集合A,假设A里面的数就是最小的。A有一个特点,维持它的最大元素,并可被查询。
2、从k+1个数到最后一个数,依次与A里的最大数top比较。如果当前数p大于等于top,说明p肯定不在A集合中,因为p比最小的k个数中最大的都大,直接跳过。如果p比top小,说明p应该在A集合中,top不应该在A中,因为现在A中除了top的数,都比top小,新来的p也比top小,所以把top赶走,把p加进来,并找到p加进来以后的最大数,当成top。循环往复,直到最后一个数查看完毕。
3、此时的A集合,就是结果,A集合里的K个数,就是最小的k个数,因为其它大的数都没有进来。
4、输出集合A里的数,从大到小,如果需要从小到大,反过来。
网友评论