题目描述
有一堆石头,每块石头的重量都是正整数。
每一回合,从中选出两块 最重的 石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块石头。返回此石头的重量。如果没有石头剩下,就返回 0。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/last-stone-weight
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
创建最大堆queue
,每次取出最大堆的前两个值(取出堆中重量最大的两块石头),并将其进行碎裂(比较这俩值)。如果碎裂后的重量不为 0,则将两数之差放进最大堆中queue
class Solution {
public int lastStoneWeight(int[] stones) {
// 创建大根堆
PriorityQueue<Integer> queue = new PriorityQueue<>((a,b) -> b-a);
for(int stone: stones){
queue.offer(stone);
}
int i = 0;
while(!queue.isEmpty() && queue.size() > 1){
int max = queue.poll();
int min = queue.poll();
if(max != min){
queue.offer(max-min);
}
i++;
}
return queue.size() == 1?queue.poll():0;
}
}
使用优先队列解决啦!