Leetcode1005K 次取反后最大化的数组和

给定一个整数数组 A,我们只能用以下方法修改该数组:我们选择某个个索引 i 并将 A[i] 替换为 -A[i],然后总共重复这个过程 K 次。(我们可以多次选择同一个索引 i。)
以这种方式修改数组后,返回数组可能的最大和。
示例 1:
输入:A = [4,2,3], K = 1
输出:5
解释:选择索引 (1,) ,然后 A 变为 [4,-2,3]。
示例 2:
输入:A = [3,-1,0,2], K = 3
输出:6
解释:选择索引 (1, 2, 2) ,然后 A 变为 [3,1,0,2]。
示例 3
输入:A = [2,-3,-1,5,-4], K = 2
输出:13
解释:选择索引 (1, 4) ,然后 A 变为 [2,3,-1,5,4]。
提示:
1 <= A.length <= 10000
1 <= K <= 10000
-100 <= A[i] <= 100

思路:
使总和最大,最大应该是再需要操作的次数K条件下:负数变为正数符号不变。负数数目n_neg ,正数数目n_pos。
负数提取形成一个序列s1 正数一个序列 正数序列求和s2
K<=a:负数从小到大排序,(排在前面绝对值反而大)
s1 = 前K个绝对值求和+[K+1:最后]直接求和
K>a:负数序列取绝对值求和s1,
K-a奇数:取出序列中最小的正数min_pos,s2 = s2 -min_pos*2(主注意这里是2倍的,最小的正数变成负数,相当于减了两次)
K-a偶数:s2不变
s = s1+s2
本以为提交成功结果出错
[-8,3,-5,-3,-5,-2]
6
**错误原因:**当K-a为奇数时候,不是取正数序列中的最小值反转,而是全部序列绝对值最小的数,因为此时负数序列都变成正数了啊!!!!!苍天啊!!!
总结思路:
如果数组全是正数,只需要反复反转最小的数,分两种情况:

K 为奇数,等同于反转 1 次
K 为偶数,等同于反转 0 次
如果数组有一部分负数,也分两种情况:

当负数个数 < 反转个数,把全部负数反转后,就成了上面全是正数的情况
当负数个数 > 反转个数,只需要尽力反转最小的负数即可得到最大值

python代码如下:

class Solution(object):
    def largestSumAfterKNegations(self, A, K):
        """
        :type A: List[int]
        :type K: int
        :rtype: int
        """
        neg = []
        pos = []
        for i in range(len(A)):
            if A[i]<0:
                neg.append(A[i])
            else:
                pos.append(A[i])
        
        s1 = abs(sum(neg))
        s2 = sum(pos)
        if K <= len(neg):
            neg.sort()
            s1 = abs(sum(neg[:K]))+sum(neg)-sum(neg[:K])
        elif (K-len(neg))%2 == 1:
            A =sorted(A,key=abs)
            s2 = s2-abs(A[0])*2
            
        s = s1+s2
        return s        
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值