'''
https://leetcode-cn.com/problems/maximize-sum-of-array-after-k-negations/
- K 次取反后最大化的數(shù)組和
給定一個整數(shù)數(shù)組 A，我們只能用以下方法修改該數(shù)組：我們選擇某個索引 i 并將 A[i] 替換為 -A[i]，然后總共重復(fù)這個過程 K
次。（我們可以多次選擇同一個索引 i。）
以這種方式修改數(shù)組后，返回數(shù)組可能的最大和。
示例 1：
輸入：A = [4,2,3], K = 1
輸出：5
解釋：選擇索引 (1,) ，然后 A 變?yōu)?[4,-2,3]。
示例 2：
輸入：A = [3,-1,0,2], K = 3
輸出：6
解釋：選擇索引 (1, 2, 2) ，然后 A 變?yōu)?[3,1,0,2]。
示例 3：
輸入：A = [2,-3,-1,5,-4], K = 2
輸出：13
解釋：選擇索引 (1, 4) ，然后 A 變?yōu)?[2,3,-1,5,4]。
提示：
1 <= A.length <= 10000
1 <= K <= 10000
-100 <= A[i] <= 100
'''
class Solution:
def largestSumAfterKNegations(self, A,K):
'''
低效方法，每次都排序
'''
for i in range(K):
A.sort()
A[0] = -A[0]
return sum(A)
def largestSumAfterKNegations1(self, A,K):
'''
使用小根堆
'''
import heapq
heapq.heapify(A)
for i in range(K):
small = heapq.heappop(A)
heapq.heappush(A,-small)
return sum(A)
def test_largestSumAfterKNegations():
s = Solution()
assert s.largestSumAfterKNegations([4,2,3],1) == 5
assert s.largestSumAfterKNegations([3,-1,0,2],3) == 6
assert s.largestSumAfterKNegations([2,-3,-1,5,-4],2) == 13
assert s.largestSumAfterKNegations1([4,2,3],1) == 5
assert s.largestSumAfterKNegations1([3,-1,0,2],3) == 6
assert s.largestSumAfterKNegations1([2,-3,-1,5,-4],2) == 13
