Leetcode1005:k次取反后最大化的数组和
-
题目:
-
给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。以这种方式修改数组后,返回数组 可能的最大和 。
-
-
思路:贪心算法
-
代码如下:
class Solution {
public int largestSumAfterKNegations(int[] nums, int k) {
//根据绝对值的大小,从大到小排序
nums = IntStream.of(nums)
.boxed()
.sorted( (o1,o2)-> Math.abs(o2) - Math.abs(o1) )
.mapToInt(Integer::intValue).toArray();
int n = nums.length;
for ( int i = 0; i < n; i++ ) {
if ( k > 0 && nums[i] < 0 ) {
nums[i] = -nums[i];
k--;
}
}
//当所有负数均变为正数后,取最小的数做剩余的反转
if ( k % 2 == 1 ) {
nums[n-1] = - nums[n-1];
}
return Arrays.stream(nums).sum();
}
}