1005.K次取反后最大化的数组和
贪心的思路,局部最优:让绝对值大的负数变为正数,当前数值达到最大,整体最优:整个数组和达到最大。
那么又是一个贪心:局部最优:只找数值最小的正整数进行反转,当前数值和可以达到最大(例如正整数数组{5, 3, 1},反转1 得到-1 比 反转5得到的-5 大多了),全局最优:整个 数组和 达到最大。
- 第一步:将数组按照绝对值大小从大到小排序,注意要按照绝对值的大小
- 第二步:从前向后遍历,遇到负数将其变为正数,同时K--
- 第三步:如果K还大于0,那么反复转变数值最小的元素,将K用完
- 第四步:求和
class Solution:
def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
nums.sort(key=lambda x:abs(x),reverse=True)#按照绝对值降序排序数组
for i in range(len(nums)):
if nums[i]<0 and k>0:
nums[i]=-nums[i]
k-=1
if k%2==1:#如果k还有剩余次数且为奇数,将绝对值最小的元素取反
nums[-1]*=-1
result=sum(nums)
return result
134. 加油站
首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈,说明 各个站点的加油站 剩油量rest[i]相加一定是大于等于零的。
每个加油站的剩余量rest[i]为gas[i] - cost[i]。
i从0开始累加rest[i],和记为curSum,一旦curSum小于零,说明[0, i]区间都不能作为起始位置,因为这个区间选择任何一个位置作为起点,到i这里都会断油,那么起始位置从i+1算起,再从0计算curSum。
那么局部最优:当前累加rest[i]的和curSum一旦小于0,起始位置至少要是i+1,因为从i之前开始一定不行。全局最优:找到可以跑一圈的起始位置。
class Solution:
def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
curSum=0#当前累计的剩余油量
totalSum=0#总剩余油量
start=0
for i in range(len(gas)):
curSum+=gas[i]-cost[i]
totalSum+=gas[i]-cost[i]
if curSum<0:#如果当前累计剩余油量小于0,起始位置更新为i+1
start=i+1
curSum=0#重新计算
if totalSum<0:
return -1
return start
135. 分发糖果
这道题目一定是要确定一边之后,再确定另一边,例如比较每一个孩子的左边,然后再比较右边,如果两边一起考虑一定会顾此失彼。
为什么不能从前向后遍历呢?
因为 rating[5]与rating[4]的比较 要利用上 rating[5]与rating[6]的比较结果,所以 要从后向前遍历。
所以确定左孩子大于右孩子的情况一定要从后向前遍历
class Solution:
def candy(self, ratings: List[int]) -> int:
candyVec=[1]*len(ratings)
#从前往后遍历,处理右比左高的情况
for i in range(1,len(ratings)):
if ratings[i]>ratings[i-1]:
candyVec[i]=candyVec[i-1]+1
for i in range(len(ratings)-2,-1,-1):
if ratings[i]>ratings[i+1]:
candyVec[i]=max(candyVec[i+1]+1,candyVec[i])
result=sum(candyVec)
return result