7.3
贪心其实很简单,每次找值最小的进行取反,如果最小的值是负数,那么可以得到最大的提升,如果最小值是非负数,那么可以得到最小的下降。
class Solution:
def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
# nums.sort()
for i in range(k):
nums.sort()
nums[0] *= -1
return sum(nums)
class Solution:
def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
for i in range(k):
min_index = nums.index(min(nums))
nums[min_index] *= -1
return sum(nums)
leetcode中进行排序要比找最小值的下标更快
随想录中是按绝对值进行降序排列,
第二步是花费次数尽可能地将其中最小的负值取反
第三步是如果还有剩余步数(此时nums是一个降序排列的全为正数的数组),那么就尽可能将最小的正值进行取反即可。
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