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

Posted 2021-09-06 数据结构和算法

截止到目前我已经写了 500多道算法题，其中部分已经整理成了pdf文档，目前总共有1000多页（并且还会不断的增加），大家可以免费下载
下载链接：https://pan.baidu.com/s/1hjwK0ZeRxYGB8lIkbKuQgQ
提取码：6666

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,堆，具体可以看下，堆中元素的添加和删除涉及到往上调整和往下调整，堆也可以看作是一棵完全二叉树，这里我随便举个例子做个视频来看一下

堆的添加

视频链接

堆的删除

视频链接