力扣刷题笔记:1005. K 次取反后最大化的数组和(时间复杂度低,思路简单、完整题解代码及注释)

题目:

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

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

示例 1:

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

示例 2:

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

示例 3:

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

提示:

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

解题思路:

要使返回的数组值最大,在K足够的情况下尽可能地把小的负数变成整数,之后若K还有剩余,则判断K剩余的奇偶:
1、若为奇数,则将A的最小值变为负值
2、若为偶数,则A无需变化

力扣刷题笔记:1005. K 次取反后最大化的数组和(时间复杂度低,思路简单、完整题解代码及注释)_第1张图片

题解python代码:

class Solution:
    def largestSumAfterKNegations(self, A: List[int], K: int) -> int:
        A.sort()
        length = len(A)
        i = 0
        while i<length and K>0:
            if A[i]<0:
                A[i] = -A[i]
                K -= 1
            else:
                if K%2==0: K = 0
                else: K = 1
                break
            i += 1
        if K==1:
            A.sort()
            A[0] = -A[0]
        return sum(A)

作者:a-qing-ge
链接:https://leetcode-cn.com/problems/maximize-sum-of-array-after-k-negations/solution/shi-jian-fu-za-du-di-si-lu-jian-dan-by-a-de3a/
来源:力扣(LeetCode)https://leetcode-cn.com/problems/maximize-sum-of-array-after-k-negations/

你可能感兴趣的:(刷题笔记,python,leetcode)