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

136 阅读1分钟

要求

给你一个整数数组 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]:
        num_freq = collections.Counter(sorted(nums, reverse=True)[ :k])
        res = []
        for x in nums:
            if num_freq[x] > 0:
                res.append(x)
                num_freq[x] -= 1
        return res

image.png

解题思路:我们先对nums进行排序取出最大的前k个元素,使用collections.Counter进行统计计数,然后循环遍历,因为我们需要保证原有的顺序不变。