[LeetCode] Maximize Sum Of Array After K Negations

Given an array A of integers, we must modify the array in the following way: we choose an i and replace A[i] with -A[i], and we repeat this process K times in total. (We may choose the same index i multiple times.)

Return the largest possible sum of the array after modifying it in this way.

Example 1:

Input: A = [4,2,3], K = 1
Output: 5
Explanation: Choose indices (1,) and A becomes [4,-2,3].

Example 2:

Input: A = [3,-1,0,2], K = 3
Output: 6
Explanation: Choose indices (1, 2, 2) and A becomes [3,1,0,2].

Example 3:

Input: A = [2,-3,-1,5,-4], K = 2
Output: 13
Explanation: Choose indices (1, 4) and A becomes [2,3,-1,5,4].

Note:

1 <= A.length <= 10000
1 <= K <= 10000
-100 <= A[i] <= 100

解题思路

要让总和最大,需要每次都取反最小的元素。一个优化点是:如果最小的元素是正数,可以直接得到最终结果,无需每次都遍历找到最小元素(最小正数取反之后还是最小的)。

实现代码

实现1:

//Runtime: 2 ms, faster than 92.17% of Java online submissions for Maximize Sum Of Array After K Negations.
//Memory Usage: 38 MB, less than 100.00% of Java online submissions for Maximize Sum Of Array After K Negations.
class Solution {
    public int largestSumAfterKNegations(int[] A, int K) {
        int sum = getSum(A);
        while (K-- > 0) {
            int index = getIndexOfSmallest(A);
            A[index] = -A[index];
            sum += 2 * A[index];
        }
        return sum;
    }

    public int getSum(int[] nums) {
        int result = 0;
        for (int num : nums) {
            result += num;
        }
        return result;
    }

    public int getIndexOfSmallest(int[] nums) {
        int smallest = Integer.MAX_VALUE;
        int index = -1;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] < smallest) {
                smallest = nums[i];
                index = i;
            }
        }
        return index;
    }
}

实现2:

//Runtime: 1 ms, faster than 99.93% of Java online submissions for Maximize Sum Of Array After K Negations.
//Memory Usage: 38 MB, less than 100.00% of Java online submissions for Maximize Sum Of Array After K Negations.
class Solution {
    public int largestSumAfterKNegations(int[] A, int K) {
        int sum = getSum(A);
        while (K > 0) {
            int index = getIndexOfSmallest(A);
            if (A[index] < 0) {
                K--;
                A[index] = -A[index];
                sum += 2 * A[index];
            } else {
                if (K % 2 == 0) {
                    return sum;
                } else {
                    return sum - 2 * A[index];
                }
            }
        }
        return sum;
    }

    public int getSum(int[] nums) {
        int result = 0;
        for (int num : nums) {
            result += num;
        }
        return result;
    }

    public int getIndexOfSmallest(int[] nums) {
        int smallest = Integer.MAX_VALUE;
        int index = -1;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] < smallest) {
                smallest = nums[i];
                index = i;
            }
        }
        return index;
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容