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

前言

题目：1005. K 次取反后最大化的数组和

参考题解：K次取反后最大化的数组和-代码随想录

---

提交代码

核心思路：尽可能将负数翻转。如果负数全部翻转之后，还有剩余翻转次数。全部用于翻转绝对值最小的那个数字。

方法一：自然数排序

按照自然数排序，实现上面思路。比较繁琐，注意细节。

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end()); // 从低到高排序

        int zero_lower_bound = lower_bound(nums.begin(),nums.end(),0) - nums.begin(); // 返回第一个大于等于0的位置

        if(k < zero_lower_bound+1){ // 全部用于将将负数变为整数
            for(int i=0; i<k; i++)
                nums[i] = -nums[i];
        }else if(k > zero_lower_bound){
            for(int i=0; i<zero_lower_bound; i++) // 将全部负数变为整数之后
                nums[i] = -nums[i];
            k = k-(zero_lower_bound+1-1); // 剩余翻转次数
            if(k%2 == 0) // 剩余为偶数，正反抵消
                ; 
            else{ // 剩余为奇数，用一次翻转当前最小的元素
                if(zero_lower_bound == 0) // 原来的元素全为整数，翻转第一个元素
                    nums[0] = -nums[0];
                else if(zero_lower_bound == nums.size()) // 原来的元素全为负数，翻转最后一个元素
                    nums[zero_lower_bound -1] = -nums[zero_lower_bound-1];
                else{ // 原来正负交替，翻转zero_lower_bound-1 or zero_lower_bound位置元素
                    int min_i;
                    if(nums[zero_lower_bound-1] < nums[zero_lower_bound])
                        min_i = zero_lower_bound-1;
                    else 
                        min_i = zero_lower_bound;
                    nums[min_i] = -nums[min_i];
                }
            }
        }

        // 累加求和
        int result = 0;
        for(int num : nums)
            result += num;
        return result;
    }
};

方法二：绝对值排序

下面思路和代码实现，来自参考题解。

• 第一步：将数组按照绝对值大小从大到小排序，注意要按照绝对值的大小
• 第二步：从前向后遍历，遇到负数将其变为正数，同时K–
• 第三步：如果K还大于0，那么反复转变数值最小的元素，将K用完
• 第四步：求和

class Solution {
static bool cmp(int a, int b) {
    return abs(a) > abs(b);
}
public:
    int largestSumAfterKNegations(vector<int>& A, int K) {
        sort(A.begin(), A.end(), cmp);       // 第一步
        for (int i = 0; i < A.size(); i++) { // 第二步
            if (A[i] < 0 && K > 0) {
                A[i] *= -1;
                K--;
            }
        }
        if (K % 2 == 1) A[A.size() - 1] *= -1; // 第三步
        int result = 0;
        for (int a : A) result += a;        // 第四步
        return result;
    }
};