提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
一、题目要求
给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后,返回数组 可能的最大和 。
示例 1:
输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3] 。
示例 2:
输入:nums = [3,-1,0,2], k = 3
输出:6
解释:选择下标 (1, 2, 2) ,nums 变为 [3,1,0,2] 。
示例 3:
输入:nums = [2,-3,-1,5,-4], k = 2
输出:13
解释:选择下标 (1, 4) ,nums 变为 [2,3,-1,5,4] 。
提示:
1 <= nums.length <= 10000
-100 <= nums[i] <= 100
1 <= k <= 10000
二、解题思路
1.排序
因为数字大小的范围为 -100 <= nums[i] <= 100 。这里可以先将负数的部分转化为正数。再将所有数字存入哈希表。
2.先修改负数部分
按照题意,要求输出的和最大且只能修改k次。因此,我们可以从最小的负数开始修改(直到k == 0 || 负数全部修改完成)。
每修改一次,k都要减去1。并且要将修改后的正数存入哈希表
3.修改负数之后k>0
如果修改完负数部分之后,k的值仍然大于0,那么我们不得不去修改非负数(即正数或者 0)了。
3.1当存在0时
无论k是奇数还是偶数,我们都可以通过不断修改0,将剩下的k全部消耗完。
3.2当不存在0时
当剩余的修改次数是偶数
由于对同一个数修改两次等价于不进行修改。因此我们也可以在不减小数组的和的前提下,把修改次数用完;
当剩余的修改次数是奇数
那么我们必然需要使用单独的一次修改将一个正数变为负数(剩余的修改次数为偶数,就不会减小数组的和)。为了使得数组的和尽可能大,我们就选择那个最小的正数。
需要注意的是,在之前将负数修改为正数的过程中,可能出现了(相较于原始数组中最小的正数)更小的正数,这一点不能忽略。
3.3返回所需数值
按照下面步骤4即可。
4.修改负数之后k<=0时
我们只需要将哈希表中存入的数字转化回来,最后返回所有数相加之和即可。
三、参考C代码
int largestSumAfterKNegations(int* nums, int numsSize, int k){
int Hash[201] = { 0 }, ret = 0;
//1.排序(先转正数)
for (int i = 0; i < numsSize; i++) {
Hash[ nums[i] + 100 ]++;
}
//2.先修改负数部分
for (int i = 0; i < 100; i++) {
while (k > 0 && Hash[i]) {
Hash[i]--; k--;
Hash[200 - i]++;
}
}
// 3.修改负数之后k<=0时 或者存在0 或者 剩下的k为偶数,
//转化数字,返回所有数字之和。
if (k <= 0 || Hash[100] || k % 2 == 0) {
for (int i = 0; i < 201; i++) {
if (Hash[i]) {
ret += Hash[i] * (i - 100);
}
}
} else { //修改负数之后k>0 && k为奇数
//寻找最小的正数
for (int i = 100; i < 201; i++) {
if (Hash[i]) {
ret -= i - 100;
Hash[i]--;
break;
}
}
//转化数字,返回所有数字之和。
for (int i = 100; i < 201; i++) {
while (Hash[i]) {
ret += i - 100;
Hash[i]--;
}
}
}
return ret;
}