学习记录@代码随想录:贪心算法part03

文章介绍了如何使用贪心算法解决LeetCode中的两个问题:1005.K次取反后最大化数组和和134.加油站问题。关键策略是根据数组绝对值排序、局部最优到整体最优,以及在分发糖果问题中考虑相邻孩子评分差异。
摘要由CSDN通过智能技术生成

学习记录@代码随想录:贪心算法part03

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

给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后,返回数组 可能的最大和 。
输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3]

题目链接:https://leetcode.cn/problems/maximize-sum-of-array-after-k-negations/
贪心的思路,局部最优:让绝对值大的负数变为正数,当前数值达到最大,整体最优:整个数组和达到最大。
那么本题的解题步骤为:
第一步:将数组按照绝对值大小从大到小排序,注意要按照绝对值的大小
第二步:从前向后遍历,遇到负数将其变为正数,同时K–
第三步:如果K还大于0,那么反复转变数值最小的元素,将K用完
第四步:求和

class Solution(object):
    def largestSumAfterKNegations(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        #key=lambda x:abs(x)指定了排序的键值函数,使用匿名函数lambda计算每个元素的绝对值。
        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
        if k%2==1:#如果k还有剩余,把绝对值最小的数乘-1
            nums[-1]*=-1
        result=sum(nums)
        return result

134. 加油站

在一条环路上有 n 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。
你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas 和 cost ,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,
否则返回 -1 。如果存在解,则 保证 它是 唯一 的。
输入: gas = [1,2,3,4,5], cost = [3,4,5,1,2]
输出: 3
解释:3 号加油站(索引为 3)出发,可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站,此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站,此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站,此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站,此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站,你需要消耗 5 升汽油,正好足够你返回到 3 号加油站。
因此,3 可为起始索引。

直接从全局进行贪心选择,情况如下:
情况一:如果gas的总和小于cost总和,那么无论从哪里出发,一定是跑不了一圈的
情况二:rest[i] = gas[i]-cost[i]为一天剩下的油,i从0开始计算累加到最后一站,如果累加没有出现负数,说明从0出发,油就没有断过,那么0就是起点。
情况三:如果累加的最小值是负数,汽车就要从非0节点出发,从后向前,看哪个节点能把这个负数填平,能把这个负数填平的节点就是出发节点。

#贪心1
class Solution(object):
    def canCompleteCircuit(self, gas, cost):
        """
        :type gas: List[int]
        :type cost: List[int]
        :rtype: int
        """
        curSum = 0#当前累计的剩余油量
        minfuel = float('inf')#从七点出发,邮箱里的油量是最小值

        for i in range(len(gas)):
            rest = gas[i]-cost[i]
            curSum += rest
            if curSum<minfuel:
                minfuel = curSum
        if curSum<0:
            return -1
        if minfuel>=0:
            return 0
        for i in range(len(gas)-1,-1,-1):
            rest = gas[i] - cost[i]
            minfuel += rest
            if minfuel >= 0:
                return i
        return -1

不太理解这种方法

#暴力
class Solution(object):
    def canCompleteCircuit(self, gas, cost):
        """
        :type gas: List[int]
        :type cost: List[int]
        :rtype: int
        """
        for i in range(len(cost)):
            rest = gas[i] - cost[i]#记录剩余油量
            index = (i+1)%len(cost)#下一个加油站的索引

            while rest>0 and index!=i:
                rest+=gas[index]-cost[index]#模拟以i为起点形式一圈
                index = (index+1)%len(cost)#更新下一个加油站的索引
            if rest>=0 and index ==i:
                return i
        return -1

135. 分发糖果

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
输入:ratings = [1,0,2]
输出:5
解释:你可以分别给第一个、第二个、第三个孩子分发 212 颗糖果。

题目链接:https://leetcode.cn/problems/candy/
在这里插入图片描述
一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
一次是从右到左遍历,只比较左边孩子评分比右边大的情况。
这样从局部最优推出了全局最优,即:相邻的孩子中,评分高的孩子获得更多的糖果。

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 i in range(len(ratings)-2,-1,-1):
            if ratings[i]>ratings[i+1]:
                candyvec[i] = max(candyvec[i],candyvec[i+1]+1)

        result = sum(candyvec)
        return result

参考链接:https://programmercarl.com/0135.%E5%88%86%E5%8F%91%E7%B3%96%E6%9E%9C.html#%E5%85%B6%E4%BB%96%E8%AF%AD%E8%A8%80%E7%89%88%E6%9C%AC

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值