业务场景:我有一个表里面有大量的数据,我查出一个List存储我的数据,我能查出每个点之间的距离,但是我要去查出每个点跟其他的点距离前三的点,算出平均值。
首先的思路就是,使用两个循环遍历,计算出当前点与集合内所有的其他的点的距离,然后再去排序,取前三个的点再算平均值。
这个思路是能实现的,但是这样的代码运算很慢,数据量一多就会有很大的性能瓶颈。
这个时候就是得使用队列,PriorityQueue优先队列,如以下代码,我们设置一个优先队列只能存三个对象,并且用distanceToCenter进行排序,优先队列会设置优先级,会根据Comparator.comparingDouble(Cell::getDistanceToCenter)返回值的大小,设置优先级,需要注意的是,越小的优先级越高,如果三个对象都没存满,那就先存满三个,如果有distanceToCenter更加小的就先清除大的(优先级低的),再将小的(优先级更高的)放进去。
PriorityQueue<Cell> queue = new PriorityQueue<>(3, Comparator.comparingDouble(Cell::getDistanceToCenter));
cells.forEach(cell -> {
if (queue.size() < 3) {
queue.add(cell);
} else if (cell.getDistanceToCenter() < queue.peek().getDistanceToCenter()) {
queue.poll();
queue.add(cell);
}
});