K次取反后最大化的数组和-贪心1005-python

没看答案:每次取反前要先排序,是因为取反最小的数字是局部最优解,取反次数用完后极为全局最优解。当数组只有0和正数时,可以无视次数,因为可以重复取反0。

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        for i in range(k):
            nums = sorted(nums)
            if nums[0] == 0:
                return sum(nums)
            else:
                nums[0] = -nums[0]
        
        return sum(nums)

优化版本:

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        numsSum = 0
        numsMin = 101 # 题目给定范围 -100 <= nums[i] <= 100
        nums = sorted(nums)

        for i in range(len(nums)):
            # 遍历整个数组,优先取反负数
            if nums[i] < 0 and k > 0: 
                nums[i] = -nums[i] 
                k -= 1
            numsMin = numsMin if numsMin < nums[i] else nums[i] # 记录取反后的数组最小值
            numsSum += nums[i]  # 记录取反后的数组和
		
		# 如果剩余次数为偶数或者取反后数组最小值为0,则与次数无关了
        # 如果剩余次数为奇数,则取反最小的正数
        if k % 2 == 1:
            numsSum -= numsMin * 2

        return numsSum
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值