代码随想录算法训练营第三十三天 | leetcode 1005. K 次取反后最大化的数组和，134. 加油站，135. 分发糖果)

1005. K 次取反后最大化的数组和

题目：
给定一个整数数组 A，我们只能用以下方法修改该数组：我们选择某个索引 i 并将 A[i] 替换为 -A[i]，然后总共重复这个过程 K 次。（我们可以多次选择同一个索引 i。）
以这种方式修改数组后，返回数组可能的最大和。

题目链接：1005. K 次取反后最大化的数组和

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        A = sorted(nums, key = abs, reverse = True) # # 将A按绝对值从大到小排列
        for i in range(len(A)):
            if k > 0 and A[i] < 0:
                A[i] *= -1
                k -= 1
        if k > 0:
            A[-1] *= (-1) ** k # 若全部取反后，k仍不为0，将最小的数字反复取反
        return sum(A)

134. 加油站

题目：
在一条环路上有 N 个加油站，其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。
如果你可以绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1。

题目链接：134. 加油站

思路记录：
如果总油量减去总消耗大于等于0那么一定可以跑完一圈。剩油量rest[i]相加一定大于等于0。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, totalSum, start = 0, 0, 0
        for i in range(len(gas)):
            curSum += gas[i] - cost[i]
            totalSum += gas[i] - cost[i]
            if curSum < 0:
                start = i + 1 # 当前累加rest[i]和 curSum一旦小于0,更新起始位置i+1
                curSum = 0 # curSum从0开始
            
        if totalSum < 0:
            return -1
        return start

135. 分发糖果

题目：
老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。
你需要按照以下要求，帮助老师给这些孩子分发糖果：
每个孩子至少分配到 1 个糖果。
相邻的孩子中，评分高的孩子必须获得更多的糖果。
那么这样下来，老师至少需要准备多少颗糖果呢？

题目链接：135. 分发糖果

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 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)