题目:
给你一个整数数组 nums
和一个整数 k
,按以下方法修改该数组:
- 选择某个下标
i
并将nums[i]
替换为-nums[i]
。
重复这个过程恰好 k
次。可以多次选择同一个下标 i
。
以这种方式修改数组后,返回数组 可能的最大和 。
示例:
- 输入: n u m s = [ 4 , 2 , 3 ] , k = 1 nums = [4,2,3], k = 1 nums=[4,2,3],k=1
- 输出: 5 5 5
- 解释: 选择下标 1 ,nums 变为 [4,-2,3] 。
解题思路:贪心策略
首先将数组按绝对值大小从大到小排序,然后每次选负数进行反转,将负数变为正数,这样是局部最优的。当遍历完数组后还有反转次数,即k>0
,此时k
若是偶数则不用管,因为偶数次反转依然结果不变,若是奇数的话,就反转最后一个数,也就是反转最小的数,这样带来的损失最小,也就是局部最优的。最后将所有数加和返回。
局部最优:让绝对值大的负数变为正数,当前数值达到最大;
全局最优:整个数组和达到最大。
class Solution {
public:
static bool cmp(int a, int b){
return abs(a) > abs(b);
}
int largestSumAfterKNegations(vector<int>& nums, int k) {
// 按绝对值大小比较,从大到小排序
sort(nums.begin(), nums.end(), cmp);
// 当还有反转次数,即k>0;且当前遍历的数小于0时,将其反转
for(int i = 0; i < nums.size(); i++){
if(k > 0 && nums[i] < 0){
nums[i] *= -1;
k--;
}
}
// 若还有反转次数,若为奇数,则将最小的数反转
if(k % 2 == 1) nums[nums.size() - 1] *= -1;
// 加和
int sum = 0;
for(int num : nums) sum += num;
return sum;
}
};