目录
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];
}
}
原理思路:
所有正数相加一定是最大子序列的和,所以,想要找到,次最大和,就要去掉序列中最小的正数,或者加上数组中最大的负数。
所以我们将数组中的数进行转换,将负数转换为正数,然后排序,利用最小堆遍历k次,找出最小和子序列,然后减去,被减去的,如果之前是正数,说明是去除了序列中的数,如果是之前是负数,那么就是在该序列中加上了负数。
子序列的寻找步骤,可以举一个例子:[1,2,3]
[]
[1] 对上一层用1进行添加,由于上一层没有数,所以不能替换
[1, 2] [2] 对上一层用2进行添加,或者替换最后一个数
[1, 2, 3] [1, 3] [2, 3] [3] 对上一层用3进行添加,或者替换最后一个数
这里我们想要找到第k大,那么sum就要减去第k小的值,利用小顶堆,然后由于顺序改变了,里面用long[] 记录,long[0]为该子序列总和,long[1]为该序列下一个要添加的数的位置。