没看答案:每次取反前要先排序,是因为取反最小的数字是局部最优解,取反次数用完后极为全局最优解。当数组只有0和正数时,可以无视次数,因为可以重复取反0。
class Solution:
def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
for i in range(k):
nums = sorted(nums)
if nums[0] == 0:
return sum(nums)
else:
nums[0] = -nums[0]
return sum(nums)
优化版本:
class Solution:
def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
numsSum = 0
numsMin = 101 # 题目给定范围 -100 <= nums[i] <= 100
nums = sorted(nums)
for i in range(len(nums)):
# 遍历整个数组,优先取反负数
if nums[i] < 0 and k > 0:
nums[i] = -nums[i]
k -= 1
numsMin = numsMin if numsMin < nums[i] else nums[i] # 记录取反后的数组最小值
numsSum += nums[i] # 记录取反后的数组和
# 如果剩余次数为偶数或者取反后数组最小值为0,则与次数无关了
# 如果剩余次数为奇数,则取反最小的正数
if k % 2 == 1:
numsSum -= numsMin * 2
return numsSum