力扣1005.K次取反后最大化的数组和[JS+CPP]

给定一个整数数组A，只能用以下方法修改该数组：选择某个索引i并将A[i]替换为-A[i]，总共重复这个过程K次。我们可以多次选择同一个索引i。以这种方式修改数组后，返回数组可能的最大和

本文参考代码随想录的解法，增加了一些理解，如果有友友像我一样看完代码和视频还是不太明白的，可以参考一下~
源码来自 代码随想录

分析：

• 如果数组中负数个数小于K，就选择其中最小的数字进行多次取反-其中需要先对负数取反，在对所有正值中的最小值多次取反
• 如果数组中负数个数大于等于K，就选择其中最小的数字依次取反，尽量保证数组中全是正值

总思路：

1. 让绝对值最大的负数变成正数–第一次贪心
2. 如果全变成正数之后K仍大于0，就找最小的正整数进行取反–第二次贪心
3. 让整个数组和变得最大

解题步骤：

1. 将数组按绝对值进行从大到小排序
2. 从前往后遍历，如果找到负数就进行取反，同时k–
3. k用完了就break，k还没用完且全是正数了就取最小的数字反复取反直到k=0
4. 对最终得到的数组求和

cpp

#include <iostream>
using namespace std;
#include <algorithm>

class Solution {
    static bool cmp(int a, int b) { return abs(a) > abs(b); }
    public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int result = 0;
        //按绝对值从大到小对数组排序，但不能直接对数组元素取绝对值，不能改变数组元素！！
        sort(nums.begin(), nums.end(), cmp);
        //遍历数组元素
        for (int i = 0; i < nums.size(); i++) {
            //如果有负数，最先遍历到的肯定是最大的负数，先把它变成正数。同时还要确定k值没用完
            if (nums[i] < 0 && k > 0) {
                nums[i] *= -1;
                k--;
            }
        }
        //如果k还没用完但是没负数了
        while (k > 0) {
            //反复对最后一个数取法，直到k用完
            nums[nums.size() - 1] *= -1;
            k--;
        }
        //遍历最终的数组，并求和--简便写法！！
        for (int a : nums) result += a;
        return result;
    }
};
//解法2
class Solution {
static bool cmp(int a, int b) {...}
public:
    int largestSumAfterKNegations(vector<int>& nums, int K) {
        sort(nums.begin(), nums.end(), cmp);       // 第一步
        for (int i = 0; i < nums.size(); i++) {...}
        //唯一注意这一步，就是判断剩下的这个k值是奇数还是偶数。
        //如果是奇数，只需对最小的数取一次反即可！！
        if (K % 2 == 1) nums[nums.size() - 1] *= -1; // 第三步
        
        int result = 0;
        for (int a : nums) result += a;        // 第四步
        return result;
    }
};

javascript

var largestSumAfterKNegations = function(nums, k) {

    nums.sort((a,b) => Math.abs(b) - Math.abs(a))
    
    for(let i = 0 ;i < nums.length; i++){
        if(nums[i] < 0 && k > 0){
            nums[i] = - nums[i];
            k--;
        }
    }

    // 若k还大于0,则寻找最小的数进行不断取反
    while( k > 0 ){
        nums[nums.length-1] = - nums[nums.length-1]
        k--;
    }
    //注意写法
    return nums.reduce((a, b) => {
         a + b
    })
};

// 版本二 (优化: 一次遍历)
var largestSumAfterKNegations = function(nums, k) {
    nums.sort((a, b) => Math.abs(b) - Math.abs(a)); // 排序
    let sum = 0;
    for(let i = 0; i < nums.length; i++) {
        if(nums[i] < 0 && k-- > 0) { // 负数取反（k 数量足够时）
            nums[i] = -nums[i];
        }
        sum += nums[i]; // 求和
    }
    if(k % 2 > 0) { // k 有多余的（k若消耗完则应为 -1）
        sum -= 2 * nums[nums.length - 1]; // 减去两倍的最小值（因为之前加过一次）
    }
    return sum;
};