给定一个整数数组A,只能用以下方法修改该数组:选择某个索引i并将A[i]替换为-A[i],总共重复这个过程K次。我们可以多次选择同一个索引i。以这种方式修改数组后,返回数组可能的最大和
本文参考代码随想录的解法,增加了一些理解,如果有友友像我一样看完代码和视频还是不太明白的,可以参考一下~
源码来自 代码随想录
分析:
- 如果数组中负数个数小于K,就选择其中最小的数字进行多次取反-其中需要先对负数取反,在对所有正值中的最小值多次取反
- 如果数组中负数个数大于等于K,就选择其中最小的数字依次取反,尽量保证数组中全是正值
总思路:
- 让绝对值最大的负数变成正数–第一次贪心
- 如果全变成正数之后K仍大于0,就找最小的正整数进行取反–第二次贪心
- 让整个数组和变得最大
解题步骤:
- 将数组按绝对值进行从大到小排序
- 从前往后遍历,如果找到负数就进行取反,同时k–
- k用完了就break,k还没用完且全是正数了就取最小的数字反复取反直到k=0
- 对最终得到的数组求和
cpp
#include <iostream>
using namespace std;
#include <algorithm>
class Solution {
static bool cmp(int a, int b) { return abs(a) > abs(b); }
public:
int largestSumAfterKNegations(vector<int>& nums, int k) {
int result = 0;
//按绝对值从大到小对数组排序,但不能直接对数组元素取绝对值,不能改变数组元素!!
sort(nums.begin(), nums.end(), cmp);
//遍历数组元素
for (int i = 0; i < nums.size(); i++) {
//如果有负数,最先遍历到的肯定是最大的负数,先把它变成正数。同时还要确定k值没用完
if (nums[i] < 0 && k > 0) {
nums[i] *= -1;
k--;
}
}
//如果k还没用完但是没负数了
while (k > 0) {
//反复对最后一个数取法,直到k用完
nums[nums.size() - 1] *= -1;
k--;
}
//遍历最终的数组,并求和--简便写法!!
for (int a : nums) result += a;
return result;
}
};
//解法2
class Solution {
static bool cmp(int a, int b) {...}
public:
int largestSumAfterKNegations(vector<int>& nums, int K) {
sort(nums.begin(), nums.end(), cmp); // 第一步
for (int i = 0; i < nums.size(); i++) {...}
//唯一注意这一步,就是判断剩下的这个k值是奇数还是偶数。
//如果是奇数,只需对最小的数取一次反即可!!
if (K % 2 == 1) nums[nums.size() - 1] *= -1; // 第三步
int result = 0;
for (int a : nums) result += a; // 第四步
return result;
}
};
javascript
var largestSumAfterKNegations = function(nums, k) {
nums.sort((a,b) => Math.abs(b) - Math.abs(a))
for(let i = 0 ;i < nums.length; i++){
if(nums[i] < 0 && k > 0){
nums[i] = - nums[i];
k--;
}
}
// 若k还大于0,则寻找最小的数进行不断取反
while( k > 0 ){
nums[nums.length-1] = - nums[nums.length-1]
k--;
}
//注意写法
return nums.reduce((a, b) => {
a + b
})
};
// 版本二 (优化: 一次遍历)
var largestSumAfterKNegations = function(nums, k) {
nums.sort((a, b) => Math.abs(b) - Math.abs(a)); // 排序
let sum = 0;
for(let i = 0; i < nums.length; i++) {
if(nums[i] < 0 && k-- > 0) { // 负数取反(k 数量足够时)
nums[i] = -nums[i];
}
sum += nums[i]; // 求和
}
if(k % 2 > 0) { // k 有多余的(k若消耗完则应为 -1)
sum -= 2 * nums[nums.length - 1]; // 减去两倍的最小值(因为之前加过一次)
}
return sum;
};