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

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

Leetcode

思路

本题的解题步骤为：

第一步：将数组按照绝对值大小从大到小排序，注意要按照绝对值的大小
第二步：从前向后遍历，遇到负数将其变为正数，同时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:
            nums[-1] = -nums[-1]

        return sum(nums)

复杂度分析

• 时间复杂度: O(nlogn)
• 空间复杂度: O(1)

134. 加油站

Leetcode

思路

每个加油站的剩余量rest[i]为gas[i] - cost[i]。

局部最优：当前累加rest[i]的和curSum一旦小于0，起始位置至少要是i+1，因为从i之前开始一定不行。全局最优：找到可以跑一圈的起始位置。

代码

class Solution:
    def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:
        if sum(gas) < sum(cost):
            return -1
        diff = [gas[i] - cost[i] for i in range(len(gas))]
        curSum = 0
        res = 0
        for i in range(len(diff)):
            curSum += diff[i]
            if curSum < 0:
                res = i + 1
                curSum = 0

        return res

复杂度分析

• 时间复杂度：O(n)
• 空间复杂度：O(1)

135. 分发糖果

Leetcode

思路

代码

class Solution:
    def candy(self, ratings: List[int]) -> int:
        candy = [1 for _ in range(len(ratings))]

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