力扣1005:k次取反后最大化的数组和
题目
给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
· 选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后,返回数组 可能的最大和 。
思路
这道题的关键点在于六个字:可以多次选择!
所以我们想要获得最大和我们就要在k还没到0前先从小到大的把负数变为正数,然后就有两种情况,k耗尽了或者是没有负数了。这时候我们再分情况讨论即可,如果k耗尽了那么最大和毋庸置疑就是当前数组的和,如果是没有负数了那么面对全部都是正数的数组我们就再需要分情况讨论了也就是当前k的值是偶数还奇数。如果k是偶数因为可以多次选择一个位置来取得相反数也就是-1的偶次幂,也就是1那么答案还是当前数组的和,如果k是奇数那就是一个位置的值要变成相反数了。那么选择哪个位置呢?当然是最小数。
代码
cpp
class Solution {
public:
int largestSumAfterKNegations(vector<int>& nums, int k) {
int n = nums.size();
int res = 0;
// 按升序进行排序
sort(nums.begin(), nums.end());
for (int i = 0; i < n; i++) {
// 如果有负数并且k大于0说明这个负数可以变成正数
if (nums[i] < 0 && k > 0) {
nums[i] = -nums[i];
k--;
}
res += nums[i];
}
// 再排一次,让最小的数排前面
sort(nums.begin(), nums.end());
if (k % 2 != 0) {
// 如果k为奇数我们就要让res加上最小数的相反数
// 所以res等于加上两倍的最小数,当然这个最小数不知道是正是负
res += -1 * 2 * nums[0];
}
// 如果k为偶数那么res就不用动,因为可以多次相反
// 偶数次就直接抵消了
return res;
}
};