代码随想录｜Day28｜贪心03｜1005.K次取反后最大化的数组和、134.加油站、135.分发糖果

最新推荐文章于 2024-07-15 22:10:26 发布

ZYdeKK

最新推荐文章于 2024-07-15 22:10:26 发布

阅读量317

点赞数 3

文章标签：算法 leetcode python 数据结构

本文链接：https://blog.csdn.net/EANFSKK/article/details/137154040

版权

文章讨论了在给定数组中通过取反操作最大化和的策略，包括优先处理绝对值最大的负数，以及在特定情况下（如K次取反、加油站加油问题和分发糖果的公平性）的优化方法。作者提供了相应的解决方案类实现。

摘要由CSDN通过智能技术生成

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

思路：

优先取反绝对值最大的负数
如果没有负数，不断取反绝对值最小的数，直到次数 K 耗尽

取反最小数有一个优化技巧：

如果 K 为偶数，则取反 K 次后，正负不变。
如果 K 为奇数，则取反 K 次后，正负变化。

class Solution:
    def largestSumAfterKNegations(self, nums: List[int], k: int) -> int:
        # 按照绝对值从大到小排序，确保优先反转绝对值最大的负数
        nums.sort(key=lambda x: -abs(x))
        for i in range(len(nums)):
            if nums[i] < 0 and k > 0:
                nums[i] *= -1
                k -= 1
        # 取反 绝对值最小值
        if k % 2 != 0:
            nums[-1] *= -1
        
        return sum(nums)

134.加油站

思路：我们可以计算每个加油站的剩余量 rest[i] = gas[i] - cost[i]。从 i = 0，开始累加每个加油站的 rest[i]，得到 curSum，如果 curSum < 0，则说明 [0, i] 中的所有加油站都不能作为起始点。

class Solution:
    def canCompleteCircuit(self, gas: List[int], cost: List[int]) -> int:

        curSum = 0
        totalSum = 0
        start = 0

        for i in range(len(gas)):
            curSum += gas[i] - cost[i]
            totalSum += gas[i] - cost[i]

            if curSum < 0:
                start = i + 1
                curSum = 0

        if totalSum < 0:
            return -1
        
        return start

135.分发糖果

先确定右边评分大于左边的情况（也就是从前向后遍历）
再确定左孩子大于右孩子的情况（从后向前遍历）

class Solution:
    def candy(self, ratings: List[int]) -> int:

        n = len(ratings)

        candys = [1] * n

        for i in range(1, n):
            if ratings[i] > ratings[i - 1]:
                candys[i] = candys[i - 1] + 1

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