日新刷题 - 2099. 找到和最大的长度为 K 的子序列

147 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第3天,点击查看活动详情

一 描述

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]

 

提示:

1 <= nums.length <= 1000 -105 <= nums[i] <= 105 1 <= k <= nums.length

二 分析

一开始我也是想用排序法去做这道题,后来发现排序后还要再去寻找值对应的地址,无疑增加了一定代码量。

于是我考虑利用双指针做这道题,发现题目要求是子数组而非连续数组,于是只需要一个指针即可完成。

我们通过不断判断指针对应值和已有数组中最小值的大小进行取代,遍历一遍即可获取所需数组,无需通过排序再找地址获得数组。

思路:

  1. 建立原始nums[0:k]数组。
  2. 利用单指针移动判断当前值和数组中最小值的大小。
  3. 如果大于最小值,将最小值删除后在数组最后加入指针值,由此可保证数组顺序不会被打乱。
  4. 返回最终数组。

这个方法遍历一次即可,复杂度是O(n),避免了排序。

三 答案

class Solution:
    def maxSubsequence(self, nums: List[int], k: int) -> List[int]:
        p1=k
        cou=nums[0:p1]
        while p1<len(nums):
            if nums[p1]>min(cou):
                cou.remove(min(cou))
                cou.append(nums[p1])
            p1+=1
        return cou

四 参考

找到和最大的长度为 K 的子序列 - 找到和最大的长度为 K 的子序列 - 力扣(LeetCode)