leetcode 1005 K 次取反后最大化的数组和

原创已于 2022-04-20 10:43:14 修改 · 173 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #排序算法 #算法

于 2022-03-01 11:14:55 首次发布

贪心算法专栏收录该内容

12 篇文章

订阅专栏

该博客主要介绍了如何在给定数组和整数K的情况下，通过K次改变数组中元素符号（负变正或正变负）来最大化数组的和。首先对数组进行排序，然后从前向后遍历，优先处理负数，若改变后的值大于其后一个值，索引加一。如果K还有剩余，继续反转最小的元素直至K用完。最后计算数组的总和。在数组长度为1时，根据K的奇偶性决定返回原值或相反数。

在这里插入图片描述

原题链接

第一步：将数组从小到大排序，这样最小值就是在最前面
第二步：从前向后遍历，遇到负数将其变为正数，判断这个值（最小值）后面的一个数是不是较小，若是则将idx++
第三步：如果K还大于0，那么反复转变数值最小的元素，将K用完
第四步：求和
二刷：若数组长度为1，则根据k是奇数还是偶数，返回不同的nums[0]
排序，使用for循环不处理最后一个位置，因为最后一个位置是最大的值，若值为负值，则进行反转，然后若其的值得绝对值小于当前反转之后的值，则说明当前值应该是负数且绝对值比之后的值要大，于是将index++，然后continue，来处理后面一个值
若值是非负的，则直接反转
最后统计所有值的和
根据k的大小进行遍历，若i 的位置小于最后一个位置，并且当前的index 位置的值为负值，则变号，判断一下若其后一个位置是小于变号之后的index位置，则index++，去执行后一个位置，关键：若其不是小于变号之后的，则反复反转当前的index位置

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        if(nums.length == 1) return k % 2 == 0 ? nums[0] : -nums[0];
        Arrays.sort(nums);
        int sum = 0;
        int index = 0;
        for(int i = 0; i < k; i++){
            if(i < nums.length - 1 && nums[index] < 0){
                nums[index] = -nums[index];
                if(nums[index] >= Math.abs(nums[index + 1])) index++;
                continue;
            }
            nums[index] = -nums[index];
        }
        for(int i = 0; i < nums.length; i++){
            sum += nums[i];
        }
        return sum;
    }
}






class Solution {
    public int largestSumAfterKNegations(int[] A, int K) {
        if(A.length == 1) return K % 2 == 0 ? A[0] : -A[0];
        Arrays.sort(A);
        int sum = 0;
        int idx = 0;
        for(int i = 0; i < K; i++){
            //不取最后一个位置，因为最后一个位置是最大的
            if(i < A.length - 1 && A[idx] < 0){
                A[idx] = -A[idx];
                if(A[idx] >= Math.abs(A[idx + 1])) idx++;
                //跳出循环后，由于idx已经++，则再次进入for时，判断的就是下一个位置了
                    continue; 
            }
   		//如果发现没有值是小于0的了，则为了实现k次取反，只能操作当前的值（将当前的值取反）         
            A[idx] = -A[idx];
        }
        //统计sum
        for(int i = 0; i < A.length; i++){
            sum += A[i];
        }
        return sum;
    }
}