LeetCode 2099. 找到和最大的长度为 K 的子序列 大顶堆 每日一题

问题描述

给你一个整数数组 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] 。
 

提示：

1 <= nums.length <= 1000
-105 <= nums[i] <= 105
1 <= k <= nums.length

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-subsequence-of-length-k-with-the-largest-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Java

class Solution {
    public int[] maxSubsequence(int[] nums, int k) {
        int len = nums.length;
        if(len == k) return nums;
        //大顶堆
        PriorityQueue<int[]> maxHeap = new PriorityQueue<>((a,b) -> ((int[])b)[1] - ((int[])a)[1]);
        for(int i = 0;i < len;i++){
            maxHeap.offer(new int[]{i,nums[i]});
        }
        //取出前k个数字
        int[][] temp = new int[k][2];
        for(int i = 0;i < k;i++){
            temp[i][0] = maxHeap.peek()[0];
            temp[i][1] = maxHeap.peek()[1];
            maxHeap.poll();
        }
        //按照索引从小到大的顺序排序
        Arrays.sort(temp,(a,b) -> ((int[])a)[0] - ((int[])b)[0]);

        int[] ans = new int[k];
        for(int i = 0;i < k;i++){
            ans[i] = temp[i][1];
        }

        return ans;
    }
}