LeetCode-in-Java
2099. Find Subsequence of Length K With the Largest Sum
Easy
You are given an integer array nums and an integer k. You want to find a subsequence of nums of length k that has the largest sum.
Return any such subsequence as an integer array of length k.
A subsequence is an array that can be derived from another array by deleting some or no elements without changing the order of the remaining elements.
Example 1:
Input: nums = [2,1,3,3], k = 2
Output: [3,3]
Explanation: The subsequence has the largest sum of 3 + 3 = 6.
Example 2:
Input: nums = [-1,-2,3,4], k = 3
Output: [-1,3,4]
Explanation: The subsequence has the largest sum of -1 + 3 + 4 = 6.
Example 3:
Input: nums = [3,4,3,3], k = 2
Output: [3,4]
Explanation: The subsequence has the largest sum of 3 + 4 = 7.
Another possible subsequence is [4, 3].
Constraints:
1 <= nums.length <= 1000-105 <= nums[i] <= 1051 <= k <= nums.length
Solution
import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;
public class Solution {
public int[] maxSubsequence(int[] nums, int k) {
// Create proirity queue with min priority queue so that min element will be removed first,
// with index
// Add those unique index in a set
// Loop from 0 to n-1 and add element in result if set contains those index
// For ex. set has index 3,5,6 Just add those element. Order will be maintained
// We are defining the min priority queue
PriorityQueue<int[]> q = new PriorityQueue<>((a, b) -> (a[0] - b[0]));
// Add element with index to priority queue
for (int i = 0; i < nums.length; i++) {
q.offer(new int[] {nums[i], i});
if (q.size() > k) {
q.poll();
}
}
// Set to keep index
Set<Integer> index = new HashSet<>();
// At the index in the set since index are unique
while (!q.isEmpty()) {
int[] top = q.poll();
index.add(top[1]);
}
// Final result add here
int[] result = new int[k];
// Just add the element in the result for those index present in SET
int p = 0;
for (int i = 0; i < nums.length; i++) {
if (index.contains(i)) {
result[p] = nums[i];
++p;
}
}
return result;
}
}