https://leetcode.cn/problems/maximize-sum-of-array-after-k-negations/
题目要求
给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
- 选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。以这种方式修改数组后,返回数组 可能的最大和 。
贪心:对数组元素的绝对值排序
public int largestSumAfterKNegations(int[] nums, int k) {
int n = nums.length;
int res = 0;
Arrays.sort(nums);
if (nums[0] > 0) {
if (k % 2 != 0) nums[0] = -nums[0];
} else {
int j = 0;
while (j < n && nums[j] < 0) j++;
for (int i = 0; i < k && i < j; i++)
nums[i] = -nums[i];
Arrays.sort(nums);
if (k > j && (k - j) % 2 != 0) nums[0] = -nums[0];
}
for (int num : nums)
res += num;
return res;
}
- 先对数组进行排序,让绝对值大的负数变为正数,当前数值达到最大;如果将负数都转变为正数了,K依然大于0,贪心:局部最优:只找数值最小的正整数进行反转,当前数值和可以达到最大。
贪心:对数组元素的绝对值排序
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 len = nums.length;
for (int i = 0; i < len; i++) {
//从前向后遍历,遇到负数将其变为正数,同时K--
if (nums[i] < 0 && K > 0) {
nums[i] = -nums[i];
K--;
}
}
// 如果K还大于0,那么反复转变数值最小的元素,将K用完
if (K % 2 == 1) nums[len - 1] = -nums[len - 1];
return Arrays.stream(nums).sum();
}
}
- 将数组按照绝对值大小从大到小排序,**注意要按照绝对值的大小,**遇到负数将其变为正数,同时K–,如果K还大于0,那么反复转变数值最小的元素,将K用完;求和。
- 因为对绝对值排序有很多流操作,所以虽然只用一次排序,但是执行时间反而比两次排序的时间长;