Leetcode-2099. 找到和最大的长度为 K 的子序列

链接

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
• -10e5 <= nums[i] <= 10e5
• 1 <= k <= nums.length

思路

题目思路很简单，就是要找出前k大的元素，然后按照这k个元素在数组中的原始顺序，存入新的数组中返回，难就难在如何按原来的顺序放进新数组。这里我们使用一个辅助的二维数组map，第一个维度存储原始的下标，第二个维度存储对应的值，如nums = [3,4,3,3]，存成map = [[0,3],[1,4],[2,3],[3,3]],如此进行两次排序即可，第一次排序对第二个维度，也就是值进行降序排序，然后再对排序后的前k个元素，按照第一个维度（下标）进行排序，再依次存入新数组中即可。这里有个很关键的知识点就是java中的sort排序函数，sort(T[] a, int formIndex, int toIndex, Comparator<? supre T> c)中，可以指定formIndex和toIndex两个参数，也就是指定排序的范围。

Java Code

class Solution {
    public int[] maxSubsequence(int[] nums, int k) {
        int n = nums.length;
        int[][] map = new int [n][2];
        for(int i=0; i<n; i++){
            map[i][0] = i;
            map[i][1] = nums[i];
        }
        Arrays.sort(map, (a, b) -> b[1] - a[1]); //对值进行降序排序
        Arrays.sort(map, 0, k, (a, b) -> a[0] - b[0]); //注意这里只对前k个进行小标升序排序

        int[] res = new int[k];
        for(int i = 0; i < k; i++){
            res[i] = map[i][1];
        }
        
        return res;

    }
}