例题:找到和最大的长度为 K 的子序列
以力扣面试题 2099题为例:
给你一个整数数组 nums 和一个整数 k 。你需要找到 nums 中长度为 k 的 子序列 ,且这个子序列的 和最大 。
请你返回 任意 一个长度为 k 的整数子序列。
子序列 定义为从一个数组里删除一些元素后,不改变剩下元素的顺序得到的数组。
示例 1:
输入:nums = [2,1,3,3], k = 2
输出:[3,3]
解释:
子序列有最大和:3 + 3 = 6 。
示例 2:
输入:nums = [-1,-2,3,4], k = 3
输出:[-1,3,4]
解释:
子序列有最大和:-1 + 3 + 4 = 6 。
示例 3:
输入:nums = [3,4,3,3], k = 2
输出:[3,4]
解释:
子序列有最大和:3 + 4 = 7 。
另一个可行的子序列为 [4, 3] 。
class Solution {
public int[] maxSubsequence(int[] nums, int k) {
int[] res = new int[k];
PriorityQueue<Pair> queue = new PriorityQueue<>(k, (a, b) -> a.value - b.value);
for (int i = 0; i < k; i++) {
queue.add(new Pair(i, nums[i]));
}
for (int i = k; i < nums.length; i++) {
if (queue.peek().value > nums[i]) {
continue;
}
queue.poll();
queue.add(new Pair(i, nums[i]));
}
PriorityQueue<Pair> queue2 = new PriorityQueue<>(k, (a, b) -> a.index - b.index);
while (!queue.isEmpty()) {
queue2.add(queue.poll());
}
int idx = 0;
while (!queue2.isEmpty()) {
res[idx] = queue2.poll().value;
idx++;
}
return res;
}
class Pair {
int index;
int value;
public Pair(int index, int value) {
this.index = index;
this.value = value;
}
}
}
评论区