给你一个整数数组 nums
和一个整数 k
,按以下方法修改该数组:
- 选择某个下标
i
并将nums[i]
替换为-nums[i]
。
重复这个过程恰好 k
次。可以多次选择同一个下标 i
。
以这种方式修改数组后,返回数组 可能的最大和
思路:
对于k次循环,有负数则最小的负数变成正数,没有负数则最小的数取反。k次循环结束后,数组之和就是要求的结果。
优化:当没有负数的时候,看此时的k的奇偶性,如果k为奇,则表示对同一个数进行取反k次后,还是原本数的负数。若k为偶,则表示取反k次还是原本的数,因此可以直接一次得到结果,而不是k次。
class Solution {
public:
int largestSumAfterKNegations(vector<int>& nums, int k) {
sort(nums.begin(),nums.end());
int i=0;
while(k--)
{
if (i < nums.size()&&nums[i] <= 0)
{
nums[i]*=-1;
i++;
}
else
{
if(k%2==0)
{
sort(nums.begin(),nums.end());
nums[0]*=-1;
}
break;
}
}
int ret=0;
for(auto e:nums)
{
ret+=e;
}
return ret;
}
};