代码随想录算法Day34(1)||贪心算法-LeetCode1005.K次取反后最大化的数组和

学习内容参考卡哥代码随想录，有文字学习资料（代码随想录网站）和视频讲解（b站）

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

题目

力扣题目链接(opens new window)

给定一个整数数组 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

学习笔记：

1.如何可以让数组和最大呢？

贪心思路：局部最优：让绝对值最大的负数变为正数，当前数值达到最大；整体最优：整个数组和达到最大。

那么如果将负数都转变为正数了，K依然大于0，此时的问题是一个有序正整数序列，如何转变K次正负，让数组和达到最大。

那么又是一个贪心：局部最优：只找数值最小的正整数进行反转，当前数值和可以达到最大（例如正整数数组{5, 3, 1}，反转1得到-1比反转5得到-5大多了），全局最优：整个数组和最大。

class Solution:
    def largestSumAfterKNegations(self, A: List[int], K: int) -> int:
        A.sort(key = lambda x: abs(x), reverse = True) # 第一步：按照绝对值降序排序数组A

    for i in range(len(A)): # 第二步：执行K次取反操作
        if A[i] < 0 and K > 0:
            A[i] *= -1
            K -= 1

    if K % 2 == 1: # 第三步：如果K还有剩余次数，将绝对值最小的元素取反
        A[-1] *= -1
    
    result = sum(A) # 第四步：计算数组A的元素和
    return result

最后

这道题的思路也很清晰了

大家一定要自己手敲代码掌握哦