每日OJ题_贪心算法二③_力扣1005. K 次取反后最大化的数组和

目录

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

解析代码

力扣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 <= 10^4
• -100 <= nums[i] <= 100
• 1 <= k <= 10^4

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {

    }
};

解析代码

贪心策略：可以先排序，然后分情况讨论，设整个数组中负数的个数为 m 个：

• m > k ：把前 k 小负数，全部变成正数；
• m == k ：把所有的负数全部转化成正数；
• m < k ：

1. 先把所有的负数变成正数。
2. 然后根据 k - m 的奇偶分情况讨论：如果是偶数，直接忽略，如果是奇数，挑选当前数组中最小的数，变成负

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int sz = nums.size(), Min = INT_MAX, sum = 0, m = 0; // m统计0和负数
        sort(nums.begin(), nums.end());
        for(int i = 0; i < sz; ++i)
        {
            sum += nums[i];
            Min = min(Min, abs(nums[i])); // 绝对值最小的数
            if(nums[i] <= 0)
                ++m;
        }
        if(m >= k)
        {
            for(int i = 0; i < k; ++i)
                sum -= 2 * nums[i]; // 最小的负数变正数，两倍
        }
        else // m < k
        {
            for(int i = 0; i < m; ++i) // 先把所有的负数变成正数
                sum -= 2 * nums[i];
            if((k - m) % 2 == 1) // 绝对值最小的数取反
                sum += 2 * -Min;
        }
        return sum;
    }
};