Leetcode 2099. Find Subsequence of Length K With the Largest Sum [Python]

Case_CaiNiao

于 2022-08-27 23:05:56 发布

阅读量171

点赞数

分类专栏： Leetcode学习记录文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/sinat_30403031/article/details/126563840

版权

Leetcode学习记录专栏收录该内容

268 篇文章 1 订阅

订阅专栏

先用array里的数字的负值和其index组成subarray，放到一个heap里，从heap里往外弹出k个subarray。依据index对k个subarray排序。然后按顺序取负值的负值。

class Solution:
    def maxSubsequence(self, nums: List[int], k: int) -> List[int]:
        container = []
        for i,nums in enumerate(nums):
            heapq.heappush(container, (-nums, i))
        tempres = []
        for _ in range(k):
            tempres.append(heapq.heappop(container))
        tempres.sort(key = lambda x:x[1])
        res = []
        for num, pos in tempres:
            res.append(-num)
        return res