代码随想录算法训练营Day 34| 贪心算法 part03 | 1005.K次取反后最大化的数组和 、134. 加油站、135. 分发糖果
文章目录
1005.K次取反后最大化的数组和
一、贪心
class Solution(object):
def largestSumAfterKNegations(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
nums.sort(key=lambda x:abs(x))
for i in range(len(nums)-1,-1,-1):
if nums[i]<0 and k>0:
nums[i] = -nums[i]
k -=1
if k ==0:
break
if k%2==1:
nums[0] = - nums[0]
print(nums)
return sum(nums)
贪心的思路,局部最优:让绝对值大的负数变为正数,当前数值达到最大,整体最优:整个数组和达到最大。
局部最优可以推出全局最优。
那么如果将负数都转变为正数了,K依然大于0,此时的问题是一个有序正整数序列,如何转变K次正负,让 数组和 达到最大。
那么又是一个贪心:局部最优:只找数值最小的正整数进行反转,当前数值和可以达到最大(例如正整数数组{5, 3, 1},反转1 得到-1 比 反转5得到的-5 大多了),全局最优:整个 数组和 达到最大。
第一步:将数组按照绝对值大小从大到小排序,注意要按照绝对值的大小
第二步:从前向后遍历,遇到负数将其变为正数,同时K–
第三步:如果K还大于0,那么反复转变数值最小的元素,将K用完
第四步:求和
134. 加油站
一、贪心
class Solution(object):
def canCompleteCircuit(self, gas, cost):
"""
:type gas: List[int]
:type cost: List[int]
:rtype: int
"""
cursum=0
totalsum = 0
start = 0
for i in range(len(gas)):
cursum += gas[i]-cost[i]
if cursum<0:
cursum=0
start = i+1
totalsum += gas[i]-cost[i]
if totalsum <0:
return -1
else:
return start
135. 分发糖果
困难!!
一、贪心
class Solution(object):
def candy(self, ratings):
"""
:type ratings: List[int]
:rtype: int
"""
result =[1]*len(ratings)
#右边的比左边的大,则右边+1
for i in range(len(ratings)-1):
if ratings[i]<ratings[i+1]:
result[i+1] = result[i]+1
# print(result)
#左边比右边大则左边+1
for i in range(len(ratings)-2,-1,-1):
if ratings[i]>ratings[i+1]:
result[i] = max(result[i+1]+1,result[i])
# print(result)
return sum(result)