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

文档讲解：代码随想录

视频讲解：代码随想录B站账号

状态：看了视频题解和文章解析后做出来了

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

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        for i in range(k):
            nums[nums.index(min(nums))] = -min(nums)
        return sum(nums)

这道题确实比较好想，给定一个整数数组和一个整数k。将数组中的任意单一元素取相反数k次，使得新数组的总和最大。

如果数组为[2,3,-4,1]，很显然第一次要取相反数的就是 -4，这样里外里就使总和增加了8.

数组变为[2, 3, 4, 1]，这时候要是再翻转，因为都是正数了，为了保证总和最大，肯定要翻转最小的数字，也就是1.

数组变为[2, 3, 4, -1]，要是再翻转，显然就是-1了，-1又刚好是当前数组内最小的数字。

通过上两轮的示例，可以发现每次翻转都翻转的是当前数组内值最小的数字。根据这个逻辑，就不难写出ac的代码了。

 时间复杂度: O(nlogn)

 空间复杂度: O(1)

134. 加油站

class Solution:
    def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
        if sum(gas) < sum(cost):
            return -1

        net_gas = []
        for i in range(len(gas)):
            net_gas.append(gas[i] - cost[i])

        cur_sum = 0
        start_index = 0
        for i in range(len(net_gas)):
            cur_sum += net_gas[i]
            if cur_sum < 0:
                cur_sum = 0
                start_index = i + 1
        return start_index

首先对比油量总和和耗油量总和，如果油量总和小于耗油量总和，那无论如何都是跑不完一圈的，直接返回-1.

用一个net_gas计算每个站点需要耗费的油量，遍历这个数组。

使用cur_sum累计到此为止油箱中的剩余油量，如果大于等于0那继续跑，但凡小于0，说明一开始的start_index不能作为起始点。start_index设置为 i + 1，也就是从下一位开始当作起始点继续模拟。

遍历完这个列表，startindex就是起始点。

时间复杂度：O(n)

空间复杂度：O(1)

135. 分发糖果

class Solution:
    def candy(self, ratings: List[int]) -> int:
        candies = [1] * len(ratings)

        for i in range(1, len(ratings)):
            if ratings[i] > ratings[i-1]:
                candies[i] = candies[i-1] + 1
        
        for i in range(len(ratings) -2 , -1 ,-1):
            if ratings[i] > ratings[i+1]:
                candies[i] = max(candies[i], candies[i+1] +1)

        return sum(candies)

这道题需要两个循环分别处理左孩子和右孩子的情况，虽然不知道为什么但就是这样的。

首先判断与左孩子的关系，如果比左孩子的ratings大，那么糖果数量是左孩子数量+1，这里用到了贪心思想，只+1.

之后再从后面向前遍历去考虑与右孩子的关系。如果不从后向前，就无法用到右孩子判断后的结果，而是只用到了和左孩子判断的结果，这样就不对了。

时间复杂度: O(n)

空间复杂度: O(n)