题目:k次取反后最大化的数组和
题解:
1)要注意贪心的思想:局部最优是先将绝对值最大的负数转换为整数,全局最优是整个数组的和最大。
2)思路是先将整个数组按照绝对值从大到小排列,先转换绝对值大负数,如果还有转换次数,则转换绝对值小的负数。
代码:
class Solution(object):
def largestSumAfterKNegations(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: int
"""
nums = sorted(nums, key=abs, reverse=True)#注意这句的写法
for i in range(len(nums)):
if nums[i] < 0 and k > 0:
k -= 1
nums[i] = nums[i] * (-1)
if k == 0:
break
if k > 0:
nums[-1] = nums[-1]* (-1)**k
return sum(nums)
题目:加油站
题解:
1)局部最优:当当前的和小于零,那么起始位置就要变成i+ 1,换一个新的起始位置。
2)首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈,说明 各个站点的加油站 剩油量rest[i]相加一定是大于等于零的。每个加油站的剩余量rest[i]为gas[i] - cost[i]。
3)i从0开始累加rest[i],和记为curSum,一旦curSum小于零,说明[0, i]区间都不能作为起始位置,因为这个区间选择任何一个位置作为起点,到i这里都会断油,那么起始位置从i+1算起,再从0计算curSum。
代码:
class Solution(object):
def canCompleteCircuit(self, gas, cost):
"""
:type gas: List[int]
:type cost: List[int]
:rtype: int
"""
rest = [0 for _ in gas]
curSum = 0
totalSum = 0
startIndex = 0
for i in range(len(gas)):
rest[i] = gas[i] - cost[i]
totalSum += rest[i]
curSum += rest[i]
if curSum < 0:#当前累加rest[i]和 curSum一旦小于0
curSum = 0
startIndex = i + 1
if totalSum < 0:
return -1
return startIndex
题目:分发糖果
题解:
本题我采用了两次贪心的策略:
一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
一次是从右到左遍历,只比较左边孩子评分比右边大的情况。
这样从局部最优推出了全局最优,即:相邻的孩子中,评分高的孩子获得更多的糖果。
代码:
class Solution(object):
def candy(self, ratings):
"""
:type ratings: List[int]
:rtype: 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 j in range(len(ratings) - 2, -1, -1):
if ratings[j] > ratings[j + 1]:
candyVec[j] = max(candyVec[j], candyVec[j + 1] + 1)
return sum(candyVec)