leetcode(力扣) 1005. K 次取反后最大化的数组和 (模拟情况分类)

题目描述:

给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:

选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

以这种方式修改数组后,返回数组 可能的最大和 。

示例 1:

输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3] 。
示例 2:

输入:nums = [3,-1,0,2], k = 3
输出:6
解释:选择下标 (1, 2, 2) ,nums 变为 [3,1,0,2] 。
示例 3:

输入:nums = [2,-3,-1,5,-4], k = 2
输出:13
解释:选择下标 (1, 4) ,nums 变为 [2,3,-1,5,4] 。

法一(繁琐):

思路分析:

这道题本身不难,但是情况比较多,容易乱。
我写了一个思路很清晰的if else代码。。。。。

首先对所给数组进行排序。
明确一点:如果k为偶数,则可以让数组中所有的数都不变换正负(拿一个数变换2次正负就行)
如果k为奇数,则只需要变换一个数的正负。

各种情况

当k为偶数时:

  • 看排序后的第0个元素是否大于0,如果大于0则直接返回数组的和。
  • 如果最左边的数小于0 ,则从左往右遍历开始将负数变为整数,执行k次,循环中当k等于0时返回数组的和。
  • 循环完如果k还是大于0则从新排序,再次判断当前k是奇数还是偶数,偶数直接返回数组的和,奇数则将最左边的最小值减去返回剩下的数的和。

当k为奇数时:

  • 直接看最左边的最小值是否大于0,如果大于0,则将最左边的最小值减去返回剩下的数的和。
  • 如果最左边的值小于0,则和k为偶数的情况一样,则从左往右遍历开始将负数变为整数,执行k次,循环中当k等于0时返回数组的和。循环完如果k还是大于0则从新排序,再次判断当前k是奇数还是偶数,偶数直接返回数组的和,奇数则将最左边的最小值减去返回剩下的数的和。

完整代码:

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        res = 0
        nums.sort()
        if k % 2 == 0:
            # 偶数的情况
            if nums[0] > 0:
                return sum(nums)
            else:
                # 有一个或更多的负值
                for p in range(len(nums)):
                    if nums[p] < 0:
                        nums[p] *= -1
                        k -=1
                        if k == 0:
                            return sum(nums)
                if k > 0:
                    nums.sort()
                    if k % 2 == 0:
                        return sum(nums)
                    else:
                        return sum(nums) - nums[0] - nums[0]
                return sum(nums)
        else:
            # 奇数的情况
            if nums[0] > 0:
                # 全部值都大于0.则值将最小的变成负数
                return sum(nums)- nums[0] - nums[0]
            else:
                # 有一个或更多的负值
                for p in range(len(nums)):
                    if nums[p] < 0:
                        nums[p] *= -1.
                        k -=1
                        if k == 0:
                            return sum(nums)
                if k > 0:
                    nums.sort()
                    if k % 2 == 0:
                        return sum(nums)
                    else:
                        return sum(nums) - nums[0] - nums[0]
                return sum(nums)

法二(简洁):

将所有数组按照绝对值进行排序。
比如’nums = [-8,3,-5,-3,-5,-2]’
排序后:[-2, 3, -3, -5, -5, -8]

此时从后向前遍历,也就是先变大的值。
当遍历到第一个值的时候,看k值剩余,偶数返回全部和,奇数返回和减去第一个值。

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        num = sorted(nums,key=abs)
        for i in range(len(num)-1,-1,-1):
            if num[i]<0:
                num[i] = num[i]*-1
                k-=1
                if k==0:
                    break
        if k!=0:
            return sum(num) if k%2==0 else sum(num[1:])-num[0]
        return sum(num)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

深度不学习!!

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值