LeetCode算法刷题(python) Day36｜08贪心算法｜1005.K次取反后最大化的数组和、134. 加油站、135. 分发糖果

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

力扣题目链接

贪心思路1：在当前数组最小的数前加负号。如果当前最小的数是负数，那么变为正数可以最大程度增大当前数值；如果当前最小数为非负数，说明当前数组已没有负数，把最小数变成负数损失最小。

然而这种做法每次变号之后要重新排序，时间复杂度为$O(kn\log n)$

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        while k > 0:
            nums.sort()
            nums[0] *= -1
            k -= 1
        return sum(nums)

贪心思路2：让绝对值大的负数变为正数，可以使得当前数值达到最大，时间复杂度$O(n\log n+k)=O(n\log n)$

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] *= -1
                k -= 1
        
        # 如果k还有剩余，（k如果为偶数，变两次负号即可抵消）
        if k % 2 == 1: 
            nums[-1] *= -1
        
        return sum(nums)

LeetCode 134. 加油站

力扣题目链接

• 首先如果总油量减去总消耗大于等于零，那么一定可以跑完一圈
• 从0开始累积每一站过后的剩余总油量，如果小于0，说明会断油，就应该从下一站出发
• 即使后面又出现负数，也不用担心，更新为后面的站就可以，因为总油量大于等于总消耗的前提一定能跑完一圈

class Solution:
    def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
        if sum(gas) < sum(cost): # 跑不完一圈
            return -1
        
        curSum = 0
        start = 0
        
        for i in range(len(gas)):
            curSum += gas[i] - cost[i]
            if curSum < 0: # 断油了，从下一站开始
                curSum = 0
                start = i + 1
        
        return start

LeetCode 135. 分发糖果

力扣题目链接

• 先初始化糖果为全1向量
• 从左向右遍历，如果右边孩子得分比左边高，result[i] = result[i -1]+1
• 此时可能有些左边孩子得分大于右边孩子的情况没有计算，此时应该从右向左遍历，因为已经保证了右孩子的最小可能，左孩子有两个选择，左孩子原本的糖果、右孩子糖果+1，应该选取两者的最大值（这是为了同时满足从左向右遍历和从右向左遍历的两种情况）

class Solution:
    def candy(self, ratings: List[int]) -> int:
        result = [1] * len(ratings)
        
        for i in range(1, len(ratings)):
            if ratings[i] > ratings[i - 1]:
                result[i] = result[i - 1] + 1
        
        for i in range(len(ratings) - 2, -1, -1):
            if ratings[i] > ratings[i + 1]:
                result[i] = max(result[i], result[i + 1] + 1)
                
        return sum(result)

今日毕！