LeetCode 1005. K 次取反后最大化的数组和

目录结构

1.题目

2.题解


1.题目

给定一个整数数组 A,我们只能用以下方法修改该数组:我们选择某个索引 i 并将 A[i] 替换为 -A[i],然后总共重复这个过程 K 次。(我们可以多次选择同一个索引 i。)

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

示例:

输入:A = [4,2,3], K = 1
输出:5
解释:选择索引 (1,) ,然后 A 变为 [4,-2,3]。


输入:A = [3,-1,0,2], K = 3
输出:6
解释:选择索引 (1, 2, 2) ,然后 A 变为 [3,1,0,2]。


输入:A = [2,-3,-1,5,-4], K = 2
输出:13
解释:选择索引 (1, 4) ,然后 A 变为 [2,3,-1,5,4]。

提示:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximize-sum-of-array-after-k-negations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2.题解

排序后,设负数个数为m,在次数K的限制下有;

  1. 若m≥K,则从前向后将m个负数取反后,计算数组元素之和返回即可;
  2. 若m
public class Solution1005 {

    @Test
    public void test1005() {
        int[] A = {2, -3, -1, 5, -4};
        int K = 5;
        System.out.println(largestSumAfterKNegations(A, K));
    }

    public int largestSumAfterKNegations(int[] A, int K) {
        Arrays.sort(A);
        int count = 0;
        for (int i = 0; i < A.length; i++) {
            if (A[i] < 0 && count < K) {
                count++;
                A[i] = -A[i];
            } else {
                break;
            }
        }
        int result = 0;
        for (int a : A) {
            result += a;
        }
        if (count < K) {
            int min = Integer.MAX_VALUE;
            for (int a : A) {
                min = Math.min(min, a);
            }
            result = (K - count) % 2 == 1 ? result - 2 * min : result;
        }
        return result;
    }
}
  • 时间复杂度:O(nlogn)
  • 空间复杂度:O(1)

你可能感兴趣的:(LeetCode,leetcode)