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

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

给你一个整数数组 nums 和一个整数 k ，按以下方法修改该数组：

选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

以这种方式修改数组后，返回数组 可能的最大和 。

示例 1：

输入：nums = [4,2,3], k = 1
输出：5
解释：选择下标 1 ，nums 变为 [4,-2,3] 。
示例 2：

输入：nums = [3,-1,0,2], k = 3
输出：6
解释：选择下标 (1, 2, 2) ，nums 变为 [3,1,0,2] 。
示例 3：

输入：nums = [2,-3,-1,5,-4], k = 2
输出：13
解释：选择下标 (1, 4) ，nums 变为 [2,3,-1,5,4] 。

提示：

1 <= nums.length <= 104
-100 <= nums[i] <= 100
1 <= k <= 104

思路讲解

其实做法就是从大到小，把负值最大的负数都取反就好了。
那就可以对数字的绝对值排序，然后遇到负值就取反，k–。
如果最后k没有用完，说明现在数组里面全是正数了，所以我们对最小的那个正数取反，把剩下的次数用完，由于取两次反等于没取，所以只需要判断k剩余的值是否是奇数就好，奇数就取一次反。

代码实现

class Solution {
public:
    //  思路：将负值最大的负数取反，不断取反即可；
    // 如果数组遍历结束之后k还是大于零就不断对最小的那个数取反，直到把k用完
    static bool cmp(int a, int b) {
        return abs(a) > abs(b);
    }
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end(), cmp); // 降序排序
        int res = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (k > 0 && nums[i] < 0) {
                nums[i] = -nums[i];
                k--;
            }
        }

        if (k % 2) nums[nums.size() - 1] *= -1; // k是奇数得取反，偶数等于不变，只要对最小的取反就好了
        for (auto i : nums)
            res += i;

        return res;
    }
};