Leetcode 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消耗完了，那么剩余的k需要对正数进行取反
对正数取反需要使用最小的正数把剩余的k消耗完

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        if (nums.length == 1) return nums[0];
        // 先对数组进行排序
        Arrays.sort(nums);
        int sum = 0;
        // 先尽可能将所有负数取反
        for (int i = 0; i < nums.length && k > 0; i++) {
            if (nums[i] < 0) {
                nums[i] = -nums[i];
                k--;
            }

        }

        // 如果k还没有消耗完，说明所有负数都已经取反了
        // 这时需要对正数进行取反
        // 如果剩余k为偶数次说明结果不变
        // 如果剩余k为奇数次说明结果需要减去最小的正数
        if (k % 2 != 0) {
            Arrays.sort(nums);
            nums[0] = -nums[0];
        }
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
        }
        return sum;
    }
}