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

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]
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值