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;
}
};