java优先队列大顶堆小顶堆
时间: 2023-11-25 18:51:49 浏览: 139
Java中的优先队列可以通过小顶堆或大顶堆来实现。小顶堆是指优先级最小的元素在队列头部,大顶堆则相反,优先级最大的元素在队列头部。在Java中,可以通过PriorityQueue类来实现优先队列,其中默认是小顶堆实现。如果需要使用大顶堆实现,则可以通过传入一个比较器来实现。以下是两种实现方式的示例代码:
1. 小顶堆实现:
```java
PriorityQueue<Integer> pq = new PriorityQueue<Integer>();
```
2. 大顶堆实现:
```java
PriorityQueue<Integer> pq = new PriorityQueue<Integer>((x,y) -> y-x);
```
关于应用方面,LeetCode题目中有一道关于优先队列的题目,即“最后一块石头的重量”。题目描述如下:有一堆石头,每块石头的重量都是正整数。每一回合,从中选出两块最重的石头,然后将它们一起粉碎。假设石头的重量分别为x和y,且x <= y。那么粉碎的可能结果如下:
- 如果x == y,那么两块石头都会被完全粉碎;
- 如果x != y,那么重量为x的石头将会完全粉碎,而重量为y的石头新重量为y-x。
最终,当只剩下一块石头时,返回这块石头的重量。可以使用优先队列来解决这个问题,具体实现方式为:将所有石头的重量加入到优先队列中,然后每次取出队列中最大的两个元素进行粉碎,将粉碎后的结果再加入到队列中,直到队列中只剩下一个元素为止。以下是示例代码:
```java
public int lastStoneWeight(int[] stones) {
PriorityQueue<Integer> pq = new PriorityQueue<Integer>((x,y) -> y-x);
for(int stone : stones) {
pq.offer(stone);
}
while(pq.size() > 1) {
int x = pq.poll();
int y = pq.poll();
if(x != y) {
pq.offer(y-x);
}
}
return pq.isEmpty() ? 0 : pq.poll();
}
```
阅读全文
相关推荐















