LeetCode 1005. Maximize Sum Of Array After K Negations

LeetCode 1005. Maximize Sum Of Array After K Negations (K 次取反后最大化的数组和)

题目

链接

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

问题描述

给你一个整数数组 nums 和一个整数 k ，按以下方法修改该数组：

选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。

以这种方式修改数组后，返回数组 可能的最大和 。

示例

输入：nums = [4,2,3], k = 1
输出：5
解释：选择下标 1 ，nums 变为 [4,-2,3] 。

提示

1 <= nums.length <= 104
-100 <= nums[i] <= 100
1 <= k <= 104

思路

首先排序一下，这里重写了比较器，用绝对值排序。

之后从绝对值大的开始计算，如果为正直接加，为副根据k加，最后处理k。

k为奇数，那就去掉两个最小数的绝对值，偶数就不用管了。

复杂度分析

时间复杂度 O(nlogn)
空间复杂度 O(n)

代码

Java

    public int largestSumAfterKNegations(int[] nums, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        for (int num : nums) {
            list.add(num);
        }
        list.sort(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return Math.abs(o2) - Math.abs(o1);
            }
        });
        int sum = 0;
        for (int i = 0; i < list.size(); i++) {
            int num = list.get(i);
            if (num >= 0) {
                sum += num;
            } else {
                if (k > 0) {
                    sum += Math.abs(num);
                    k--;
                } else {
                    sum += num;
                }
            }
        }
        k = k % 2;
        int tmp = list.get(list.size() - 1);
        if (k == 1) {
            sum -= 2 * Math.abs(tmp);
        }
        return sum;
    }