1005.K次取反后最大化的数组和
给定一个整数数组 A,我们只能用以下方法修改该数组:我们选择某个索引 i 并将 A[i] 替换为 -A[i],然后总共重复这个过程 K 次。(我们可以多次选择同一个索引 i。)
以这种方式修改数组后,返回数组可能的最大和。
示例 1:
- 输入:A = [4,2,3], K = 1
- 输出:5
- 解释:选择索引 (1,) ,然后 A 变为 [4,-2,3]。
示例 2:
- 输入:A = [3,-1,0,2], K = 3
- 输出:6
- 解释:选择索引 (1, 2, 2) ,然后 A 变为 [3,1,0,2]
本题可以操作同一个索引多次,因此
如果当前数组元素均为正数,且K为奇数时,操作最小的那个,如果为偶数,则操作谁都无所谓。
如果当前数组元素存在负数,从绝对值最大的负数开始都取反为正数,若还有剩余,剩下的若为奇数,挑最小的来进行取反,若为偶数,则无所谓。
因此可以将数组先进行排序。
import java.util.Arrays;class Solution {public int largestSumAfterKNegations(int[] nums, int k) {Arrays.sort(nums);int res = 0;int i = 0; //记录第一个非负数索引// 将负数取反,直到k用完或遇到第一个非负数while (k > 0 && i < nums.length && nums[i] < 0) {nums[i]= -nums[i];k--;i++;}Arrays.sort(nums);// 如果k为奇数且仍有剩余,则将最小的非负数取反if (k % 2 == 1 ) {nums[0]=-nums[0];}// 加上剩余的数字for(i=0; i<nums.length;i++){res += nums[i];}return res;}
}