題目鏈接
難度:簡單 ??????類型: 數(shù)組
給定一個(gè)整數(shù)數(shù)組 A,我們只能用以下方法修改該數(shù)組:我們選擇某個(gè)個(gè)索引 i 并將 A[i] 替換為 -A[i],然后總共重復(fù)這個(gè)過程 K 次。(我們可以多次選擇同一個(gè)索引 i。)
以這種方式修改數(shù)組后,返回?cái)?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.從小到大排序
2.先用取反的機(jī)會(huì)把負(fù)數(shù)變正,直到k次機(jī)會(huì)用光或者所有數(shù)都為正
3.若k次機(jī)會(huì)用光,直接求和;若沒用光,說明數(shù)組a此時(shí)都是正數(shù),若剩余的取反次數(shù)不能被2整除,必定有個(gè)數(shù)會(huì)變?yōu)樨?fù)數(shù),讓數(shù)組最小值變?yōu)樨?fù)求和可得到最小值,若能被2整除,直接求和
代碼實(shí)現(xiàn)
class Solution:
def largestSumAfterKNegations(self, A: List[int], K: int) -> int:
a = sorted(A)
i, n = 0, len(a)
while i<n and i<K and a[i]<0:
a[i] = -a[i]
i += 1
return sum(a) - (K-i)%2 *2*min(a)