0
点赞
收藏
分享

微信扫一扫

LeetCode 1046. 最后一块石头的重量

截止到目前我已经写了 500多道算法题,其中部分已经整理成了pdf文档,目前总共有1000多页(并且还会不断的增加),大家可以免费下载

LeetCode 1046. 最后一块石头的重量_java

LeetCode 1046. 最后一块石头的重量_算法_02

public int lastStoneWeight(int[] stones) {
//最大堆,也就是元素最大的在堆顶
PriorityQueue<Integer> pq = new PriorityQueue<>((a, b) -> b - a);
//把数组中的元素全部放入堆中
for (int num : stones)
pq.offer(num);
while (pq.size() > 1) {
//分别取出堆中最大的值和第二大的值
int largest = pq.poll();
int large = pq.poll();
//如果largest和large一样大,相当于他俩玉石俱焚了,
//否则就把他俩的差值放到堆中
if (largest > large)
pq.offer(largest - large);
}
//最后如果堆是空的,说明他们全部都玉石俱焚了,否则就返回
//堆中仅有的那个值
return pq.isEmpty() ? 0 : pq.poll();
}

前面也讲过​​378,数据结构-7,堆​​,具体可以看下,堆中元素的添加删除涉及到往上调整往下调整,堆也可以看作是一棵完全二叉树,这里我随便举个例子做个视频来看一下

堆的添加

LeetCode 1046. 最后一块石头的重量_java_03
​​​视频链接​​

堆的删除

LeetCode 1046. 最后一块石头的重量_java_04
​​​视频链接​​


举报

相关推荐

0 条评论