Java小根堆的使用

题目：超过阈值的最少操作数 II

思路

一眼想到用小根堆，学到了一些小根堆的相关的信息

收获

• heap.add(),添加元素
• heap.poll(),获取堆顶元素，并移除
• heap.peek(),获取堆顶元素
• heap.size(),获取堆的元素数量

代码

class Solution {public int minOperations(int[] nums, int k) {int res = 0;PriorityQueue<Integer> heap = new PriorityQueue<>();for (int num : nums) {if (num < k) heap.add(num);}while (heap.size() > 0) {res++;System.out.println(heap.size());if (heap.size() == 1) break;int m1 = heap.poll();int m2 = heap.poll();long t = (long)m1 * 2 + m2;if (t < k) {int s = (int) t;System.out.println(s);heap.add(s);}}return res;}
}