【week307-amazon】leetcode2386. 找出数组的第 K 大和

最新推荐文章于 2024-03-09 15:16:20 发布

柒年时光

最新推荐文章于 2024-03-09 15:16:20 发布

阅读量281

点赞数

分类专栏： leetcode Python 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/jqq125/article/details/126468017

版权

Python 同时被 2 个专栏收录

484 篇文章 1 订阅

订阅专栏

leetcode

461 篇文章 1 订阅

订阅专栏

题目：
给你一个整数数组 nums 和一个正整数 k 。你可以选择数组的任一子序列并且对其全部元素求和。

数组的第 k 大和定义为：可以获得的第 k 个最大子序列和（子序列和允许出现重复）

返回数组的第 k 大和。

子序列是一个可以由其他数组删除某些或不删除元素排生而来的数组，且派生过程不改变剩余元素的顺序。

注意：空子序列的和视作 0 。
在这里插入图片描述
思路：

在这里插入图片描述

解答：

class Solution:
    def kSum(self, nums: List[int], k: int) -> int :
        sum=0
        for i, x in enumerate(nums):
            if x >= 0: sum += x
            else: nums[i] = -x
        nums.sort()
        #print(nums)
        h = [(0, 0)] 
        for _ in range(k - 1):
            s, i = heappop(h) #s为待减去的子序列的和
            if i < len(nums):
                heappush(h, (s + nums[i], i + 1))  # 保留 nums[i-1]
                if i: 
                    heappush(h, (s + nums[i] - nums[i - 1], i + 1))  # 不保留 nums[i-1]
            #print(i,h)
        return sum-h[0][0]