作者: 负雪明烛
id: fuxuemingzhu
个人博客: http://fuxuemingzhu.cn/
题目地址:https://leetcode.com/problems/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
题意
对于一个由整数构成的数组 A,每次翻转可以把其中的任意一位翻转成其相反数。要求一定要做 K 次翻转,注意可以翻转相同位置的数字。求翻转 K 次之后的结果数组的可能的最大和。
解题思路
这个题目怎么想呢?
- 我们优先翻转负数翻转成正数,这样和就会变大。那么优先翻转哪个负数呢?肯定是最小的负数,这样求相反数之后会变得最大。
- 那么,当负数翻转完了之后怎么办?那么只能翻转非负数了,所以如果有 0 就一直翻转 0,否则就每次挑正数翻转成负数,翻转之后继续选负数翻转。
总之,这是个贪心算法:每次都翻转数组中最小的数字,翻转 K 次之后,得到的数组拥有最大和。
小根堆
为了快速得到数组中的最小数字,用 min()
函数是不行的,因为其时间复杂度是 O(N)
。
最好的办法是:维护一个小根堆,每次翻转前取出堆里面的最小数字,翻转之后的结果仍然放入堆中,以便进行下次翻转。
下面的动画,是根据题目示例「输入: nums = [2,-3,-1,5,-4], k = 2 输出: 13」所做。
Python代码如下:
class Solution:
def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
heapq.heapify(nums)
for _ in range(k):
curmin = heapq.heappop(nums)
heapq.heappush(nums, -curmin)
return sum(nums)
总结
- 今天的题目不难,只要想到用一个堆结构,基本就柳暗花明了。
日期
2019 年 3 月 10 日 —— 周赛进了第一页!