1005. K 次取反后最大化的数组和

地址：

力扣https://leetcode-cn.com/problems/maximize-sum-of-array-after-k-negations/

题目：

给你一个整数数组 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 <= 104 -100 <= nums[i] <= 100 1 <= k <= 104

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximize-sum-of-array-after-k-negations
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

要考虑几种情况：

1. 全正数

        如果 k 是奇数，那么就翻 1 次 最小值

        如果 k 是偶数，不考虑翻牌

2. 全负数

        如果 k 小于等于负数个数，那么把 k 个负数翻牌

        如果 k 大于负数个数，那么所有负数都翻牌后考虑 剩下的 k 的奇偶性

                剩余 k 是奇数，那么翻最末的数

                剩余 k 是 偶数，不考虑翻牌

3. 有正有负

        如果 k 小于等于负数个数，那么把 k 个负数翻牌

        如果 k 大于负数个数，那么表示负数全部变为正数了，再来考虑剩余 k 的奇偶性

                剩余 k 是奇数，最后一个负数代表的正数，于第一个正数比较，那个小翻那个

                剩余 k 是偶数，不考虑翻牌

结合这么多条件，所以代码有点啰嗦。

后面改良了下，暂时不考虑那么多，只要 k 能覆盖的情况下(k>0)，遇见负数就翻牌，翻完 k--

循环的过程把所有元素（变化完后的）累加

接着再来处理是否需要判定的情况

如果 k 有剩余且为奇数：

        前面的循环过程会记录第一个正数的 index

        如果 index 仍为初始值，表示整个数组全负，我们要减去最末元素值 * 2

        如果 index 为 0，表示整个数组全正，我们要减去首元素 * 2

        其余情况表示数组有正有负，需要判定 负数于正数界限的值那个那个小，减去其值 * 2

        这里都是要减去 *2，因为循环内已经加了一次，所以去掉加的值，再去掉最后要翻的一次值

方法一、罗列各种条件

int cmp(const void *a, const void *b)
{
	return *(int *)a - *(int *)b;
}

int largestSumAfterKNegations(int* nums, int numsSize, int k){
	int i;
	int sum = 0;
	int negnum = 0;
	int idx = -1;
	
	qsort(nums, numsSize, sizeof(int), cmp);
	
	if(nums[0] > 0)
	{
		if(k & 1)
			nums[0] = 0 - nums[0];
	}
	else
	{
		for(i=0; i<numsSize; i++)
		{
			if(nums[i] < 0)
				negnum++;
			else
			{
				idx = i;
				break;
			}
		}
		
		if(negnum >= k)
		{
			i = 0;
			while(k--)
			{
				nums[i] = 0 - nums[i];
				i++;
			}
		}
		else if(negnum == 0)
		{
			if(k & 1)
				nums[0] = 0 - nums[0];
		}
		else
		{
			int k2 = k - negnum;
			
			i = 0;
			while(negnum--)
			{
				nums[i] = 0 - nums[i];
				i++;
			}
			
			if(idx == -1)
			{
				if(k2 & 1)
					nums[numsSize-1] = 0 - nums[numsSize-1];
			}
			else
			{
				if(k2 & 1)
				{
					//printf("nums[%d]=%d, nums[%d]=%d\n", idx-1, nums[idx-1], idx, nums[idx]);
					if(nums[idx-1] <= nums[idx])
						nums[idx-1] = 0 - nums[idx-1];
					else
						nums[idx] = 0 - nums[idx];
				}
			}
		}
	}
	
	for(i=0; i<numsSize; i++)
	{
		sum += nums[i];
	}
	
	return sum;
}

方法二、先累计再判定条件

int cmp(const void *a, const void *b)
{
	return *(int *)a - *(int *)b;
}

int largestSumAfterKNegations(int* nums, int numsSize, int k){
	int i;
	int sum = 0;
	int idx = -1;
	
	qsort(nums, numsSize, sizeof(int), cmp);
	
	for(i=0; i<numsSize; i++)
	{
		if(idx == -1 && nums[i] >= 0)
			idx = i;
			
		if(k > 0 && nums[i] < 0)
		{
			nums[i] = 0 - nums[i];
			k--;
		}
		
		//printf("sum=%d + nums[%d]=%d, k=%d\n", sum, i, nums[i], k);
		sum += nums[i];
	}
	
    //printf("k=%d, idx=%d, sum=%d\n", k, idx, sum);
	if( (k > 0) && (k & 1) )
	{
		if(idx == -1)
			sum -= nums[numsSize-1] * 2;
		else if(idx == 0)
			sum -= nums[idx] * 2;
		else 
		{
			if(nums[idx-1] < nums[idx])
				sum -= nums[idx-1] * 2;
			else
				sum -= nums[idx] * 2;
		}
	}
	
	return sum;
}

查看更多刷题笔记