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

Posted on | In
Words count in article: 329 | Reading time ≈ 1

题目

给你一个整数数组 nums 和一个整数 k 。你需要找到 nums 中长度为 k 的 子序列 ，且这个子序列的 和最大 。

请你返回 任意 一个长度为 k 的整数子序列。

子序列 定义为从一个数组里删除一些元素后，不改变剩下元素的顺序得到的数组。

示例1：

1
2
3
4
输入：nums = [2,1,3,3], k = 2
输出：[3,3]
解释：
子序列有最大和：3 + 3 = 6 。

示例2：

1
2
3
4
输入：nums = [-1,-2,3,4], k = 3
输出：[-1,3,4]
解释：
子序列有最大和：-1 + 3 + 4 = 6 。

示例3：

1
2
3
4
5
输入：nums = [3,4,3,3], k = 2
输出：[3,4]
解释：
子序列有最大和：3 + 4 = 7 。
另一个可行的子序列为 [4, 3] 。

提示：

  • 1 <= nums.length <= 1000
  • -10^5 <= nums[i] <= 10^5
  • 1 <= k <= nums.length

解法

解法一：

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
public int[] maxSubsequence(int[] nums, int k) {
        if (k == nums.length) {
            return nums;
        }
        int[] newNums = new int[nums.length];
        System.arraycopy(nums, 0, newNums, 0, nums.length);
        Arrays.sort(newNums);
        HashMap<Integer, Integer> map = new HashMap<>();

        for (int i = newNums.length - k; i < newNums.length; i++) {
            map.put(newNums[i], map.getOrDefault(newNums[i], 0) + 1);
        }
        int[] res = new int[k];
        int index = 0;
        for (int num : nums) {
            if (map.getOrDefault(num, 0) > 0 && index < k) {
                map.put(num, map.get(num) - 1);
                res[index++] = num;
            }
        }
        return res;
    }
0%