LeetCode：2386. 找出数组的第 K 大和（优先级队列 Java）

2386. 找出数组的第 K 大和

题目描述：

实现代码与解析：

优先级队列（小顶堆）

原理思路：

2386. 找出数组的第 K 大和

题目描述：

给你一个整数数组 nums 和一个正整数 k 。你可以选择数组的任一 子序列 并且对其全部元素求和。

数组的 第 k 大和 定义为：可以获得的第 k 个最大子序列和（子序列和允许出现重复）

返回数组的 第 k 大和 。

子序列是一个可以由其他数组删除某些或不删除元素排生而来的数组，且派生过程不改变剩余元素的顺序。

注意：空子序列的和视作 0 。

示例 1：

输入：nums = [2,4,-2], k = 5
输出：2
解释：所有可能获得的子序列和列出如下，按递减顺序排列：
- 6、4、4、2、2、0、0、-2
数组的第 5 大和是 2 。

示例 2：

输入：nums = [1,-2,3,4,-10,12], k = 16
输出：10
解释：数组的第 16 大和是 10 。

提示：

n == nums.length
1 <= n <= 105
-109 <= nums[i] <= 109
1 <= k <= min(2000, 2n)

实现代码与解析：

优先级队列（小顶堆）

class Solution {public long kSum(int[] nums, int k) {long sum = 0; // 计算正数和int n = nums.length; // 所有正数求和，负数取反重新赋值for (int i = 0; i < n; i++) {if (nums[i] >= 0) sum += nums[i];else nums[i] = -nums[i];}// 排序Arrays.sort(nums);// 优先级队列（小顶堆） // int[] long long[0]是当前该子序列总和，long[1]是其下一个小添加的数PriorityQueue<long[]> pq = new PriorityQueue<>((a, b) -> (Long.compare(a[0], b[0])));// 先从空数组开始，pq.add(new long[]{0, 0}); // 总和， 下一个数// 遍历while (--k > 0) {long[] t = pq.peek();pq.poll();long cur = t[0];long next = t[1];// 如果下一个需要添加或替换的数的位置是n，那么跳过if (next == n) continue;// 添加下一个数，加入到堆中，更新总和和下一个位置pq.offer(new long[]{cur + nums[(int)next], next + 1});// 替换最后一个数，更新总和和下一个位置if (next != 0) {pq.offer(new long[]{cur - nums[(int)next - 1] + nums[(int)next], next + 1});}}// 当前堆顶就是最小值，sum减去return sum - pq.peek()[0];}
}