每日一练：贪心-K 次取反后最大化的数组和

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

题目要求：

给你一个整数数组 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

贪心算法 O(N)：

        假如k==1，很显然只需要使最小的元素取负即可，当k>1时，它实际上是很多个k==1组成的，故循环k次，每一次都找到最小的元素取反即可。

        需要修改数组找到最小值时，小根堆的效率是很高的。

struct com {static bool operator()(int a, int b) { return a > b; }
};
class Solution {
public:int largestSumAfterKNegations(vector<int>& nums, int k) {priority_queue<int, vector<int>, com> heap;int sum = 0;for (const int num : nums) {sum += num;heap.push(num);}while (k--) {int tmp = heap.top(); // 得到最小值heap.pop(); // 去掉最小值sum -= 2 * tmp; // 得到最小值取负的和heap.push(-tmp); // 最小值取负放入小根堆}return sum;}
};