1005 k次取反后最大化的数组和
给你一个整数数组 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]
解题思路:贪心,排序后尽可能把所有负数变成正数。如果都是正数后,k不等于0且为奇数,再排序一次,把最小值变为负数即可。
int cmp(const void *a,const void *b){
return *(int*)a-*(int*)b;
}
int largestSumAfterKNegations(int* nums, int numsSize, int k) {
qsort(nums,numsSize,sizeof(int),cmp);
for(int i=0;i<numsSize;i++){
if(nums[i]<0&&k>0){
nums[i]*=-1;
k--;
}else break;
}
if(k%2!=0){
qsort(nums,numsSize,sizeof(int),cmp);
nums[0]*=-1;
}
int sum=0;
for(int i=0;i<numsSize;i++){
sum+=nums[i];
}
return sum;
}
也可以按绝对值从大到小排序,这样循环就可以先把大的负数变成正数。最后,如果k为奇数,把最小的元素,即最后一个元素nums[n-1]变为相反数。
int abs(int a){
if(a>=0)return a;
else return -a;
}
int cmp(const void *a,const void *b){
return abs(*(int*)b)-abs(*(int*)a);
}//绝对值排序 从小到大
int largestSumAfterKNegations(int* nums, int numsSize, int k) {
qsort(nums,numsSize,sizeof(int),cmp);
for(int i=0;i<numsSize;i++){
if(nums[i]<0&&k>0){
nums[i]*=-1;
k--;
}
}
if(k%2!=0&&k>0){
nums[numsSize-1]*=-1;
}
int sum=0;
for(int i=0;i<numsSize;i++){
sum+=nums[i];
}
return sum;
}