算法-贪心篇07-K 次取反后最大化的数组和

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

力扣题目链接

题目描述

给你一个整数数组 nums 和一个整数 k ，按以下方法修改该数组：
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后，返回数组 可能的最大和 。

解题思路

对数组进行排序，优先处理最小的复数；然后有三种情况：
情况一：k比负数的数量少，直接结束；
情况二：k比负数的数量大，但是有数据为0；因为零把剩下取反的次数全部消耗完，取反不影响；
情况三：k比负数的数量大，没有数据为0；如果剩下的次数是2的倍数，结果不变。如果剩下的次数不是2的倍数，在已经变完的数据中选一个最小的取反。

最后，把数据相加得出结果。

题解

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int ans = 0;
        sort(nums.begin(), nums.end());

        for(int i = 0; i < nums.size(); i++){
            if(i < k && nums[i] < 0){
                nums[i] = -nums[i];
            }
            else if(i < k && nums[i] == 0){
                break;
            }
            else if(i < k && nums[i] > 0){
                if((k - i) % 2 == 1){
                    if(i == 0 || nums[i] < nums[i - 1]){
                        nums[i] = -nums[i];
                    }
                    else {
                        nums[i - 1] = -nums[i - 1];
                    }
                } 
                break;
            }
        }

        for(int n : nums){
            ans += n;
        }

        return ans;
    }
};