PriorityQueue 名叫优先级队列,底层由堆结构实现,默认是小根堆。通过 offer 方法添加进去的元素会进行堆排序,最小的元素放在堆顶。通过 peek 方法可以获得堆顶(最小)元素。通过 poll 方法可以删除堆顶元素同时获得堆顶元素,删除之后剩下的元素中最小的元素仍处于堆顶。
一、应用场景
某电商平台入驻了大量的商家,商家可以在平台销售商品,用户可以在平台的商家那里购买商品。用户付款后如果对购买到的商品不满意,可以向平台发起投诉。用户对某商家的某件商品的投诉记录会存储在一张表中,表结构如下:
列名 | 类型 | 备注 |
---|---|---|
store_id | int | 商家id |
product_id | int | 商品id |
remark | string | 投诉记录 |
现在的需求是:找到投诉记录最多的前 3 个商家,目的是在搜索时对其店铺进行降权处理。
二、思路分析
- 创建一个小根堆(PriorityQueue 默认就是小根堆)
- 小根堆中元素的数量小于 3 的时候就直接向集合中添加元素
- 当堆中的元素个数等于 3 的时候,通过 peek 方法取出堆顶元素(最小的那个)与当前遍历到的元素比较
- 如果当前遍历到的元素大于堆顶元素,就把原堆顶元素移除,把当前元素加入堆中
- 这样使得移除的元素都小于堆中的元素
- 所以最后堆中保留下来的就是最大的N个元素
三、代码实现
import java.util.*;
// 投诉日志实体类
class ComplainLog {
// 商家id
public Integer storeId;
// 商品id
public Integer productId;
// 投诉记录
public String remark;
ComplainLog(Integer storeId, Integer productId, String remark) {
this.storeId = storeId;
this.productId = productId;
this.remark = remark;
}
}
// 商家被投诉次数实体类
class ComplainCount {
// 商家id
public Integer storeId;
// 投诉次数
public Integer complainCount;
ComplainCount(Integer storeId, Integer complainCount) {
this.storeId = storeId;
this.complainCount = complainCount;
}
}
public class PriorityQueueTest {
public static void main(String[] args) {
// 模拟10条投诉记录
ComplainLog log1 =