python笔记(4) --- 2099. 找到和最大的长度为 K 的子序列-CSDN博客

本文链接：https://blog.csdn.net/sx_logicflow/article/details/149078378

题目来自力扣2099. 找到和最大的长度为 K 的子序列 - 力扣（LeetCode）

记录每日刷题学习过程，如有侵权问题，联系作者删除。

欢迎有意向的同学们交流学习。

一、题目描述

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

二、思路笔记

题目要求找出长度为k的子序列，使得它的和为所有子序列的最大值。同时要求子序列中数字相对位置要与原始数组保持一致。

可以依据数组元素大小，对数组的index进行排序，排序后切片获得长度为k的子序列，然后依照此时子序列中的index，依次将元素从数组中取出。

三、代码

class Solution:
    def maxSubsequence(self, nums: List[int], k: int) -> List[int]:
        idx_list = sorted(range(len(nums)), key=lambda i: nums[i])
        idx_list = sorted(idx_list[-k:])
        return [nums[i] for i in idx_list]