有界优先队列-BoundedPriorityQueue

简介

举个例子。我有一个用户表,这个表根据用户名被Hash到不同的数据库实例上,我要找出这些用户中最热门的5个,怎么做?我是这么做的:

  • 在每个数据库实例上找出最热门的5个
  • 将每个数据库实例上的这5条数据按照热门程度排序,最后取出前5条这个过程看似简单,但是你应用服务器上的代码要写不少。首先需要Query N个列表,加入到一个新列表中,排序,再取前5。这个过程不但代码繁琐,而且牵涉到多个列表,非常浪费空间。

于是,BoundedPriorityQueue应运而生。

先看Demo:

  1. /**
  2. * 有界优先队列Demo
  3. * @author Looly
  4. *
  5. */
  6. public class BoundedPriorityQueueDemo {
  7. public static void main(String[] args) {
  8. //初始化队列,设置队列的容量为5(只能容纳5个元素),元素类型为integer使用默认比较器,在队列内部将按照从小到大排序
  9. BoundedPriorityQueue<Integer> queue = new BoundedPriorityQueue<Integer>(5);
  10. //初始化队列,使用自定义的比较器
  11. queue = new BoundedPriorityQueue<>(5, new Comparator<Integer>(){
  12. @Override
  13. public int compare(Integer o1, Integer o2) {
  14. return o1.compareTo(o2);
  15. }
  16. });
  17. //定义了6个元素,当元素加入到队列中,会按照从小到大排序,当加入第6个元素的时候,队列末尾(最大的元素)将会被抛弃
  18. int[] array = new int[]{5,7,9,2,3,8};
  19. for (int i : array) {
  20. queue.offer(i);
  21. }
  22. //队列可以转换为List哦~~
  23. ArrayList<Integer> list = queue.toList();
  24. System.out.println(queue);
  25. }
  26. }

原理非常简单。设定好队列的容量,然后把所有的数据add或者offer进去(两个方法相同),就会得到前5条数据了。