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
class Solution:def maxSubsequence(self, nums: List[int], k: int) -> List[int]:n = len(nums)# 创建一个下标数组 idx=[0,1,2,…,n−1]idx = [i for i in range(n)]# lambda i: nums[i] 是一个匿名函数,表示对每个索引 i,取 nums[i] 作为排序依据# 根据 nums[idx[i]] 对 idx 排序,这样就知道最大的 k 个数在哪里了。# sorted默认升序idx = sorted(idx,key = lambda i:nums[i])# 取前k大的元素下标并按升序排序idx = sorted(idx[-k:])return [nums[i] for i in idx]