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

解答

思路

  1. 最开始以为只需要快排后取后K位即可，但是顺序不符合子序列的要求，所以最后还进行了一次排序

  2. 个人认为最快的方法是在快排的时候同时记录数组下标的变化，最后则可不进行排序了

代码

class Solution {
    public int[] maxSubsequence(int[] nums, int k) {
        int[] tempNums = new int[nums.length];
        System.arraycopy(nums, 0, tempNums, 0, nums.length);
        Arrays.sort(tempNums);
        int[] resultTemp = new int[k];
        System.arraycopy(tempNums, tempNums.length - k, resultTemp, 0, k);
        int[] result = new int[k];
        int index = 0;
        for (int num : nums) {
            for (int i = 0; i < resultTemp.length; i++) {
                if (num == resultTemp[i]) {
                    result[index++] = num;
                    resultTemp[i] = Integer.MIN_VALUE;
                    break;
                }
            }
        }
        return result;
    }
}