代码随想录训练营 贪心03

代码随想录训练营 贪心03

  • 135.分发糖果
      • 思路
        • 代码
  • 1005. K 次取反后最大化的数组和
        • 思路
        • code
  • 134. 加油站
        • 思路
        • code

135.分发糖果

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。

思路

从左到右,从右到左,分别遍历,好难想啊

代码

func candy(ratings []int) int {
    need :=make([]int,len(ratings))
    sum:=0
    //初始化 一人以恶搞
    for i:=0;i <len(ratings);i++{need[i] =1}
    //左到右 左比右小 右加1
    for i:=0;i<len(ratings)-1;i++{
        if ratings[i+1]>ratings[i]{
            need[i+1] = need[i]+1
        }
    }
    //从右到左 左比右大 左+1
    for i:=len(ratings)-1;i>0;i--{
        if ratings[i-1] > ratings[i]{
            need[i-1] = max(need[i-1], need[i]+1)
        }
    }
    for _,v:=range need{
        sum+=v
    }
    return sum
}

func max(num1 int, num2 int) int {
    if num1 > num2 {
        return num1
    }
    return num2
}

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

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

思路

排序 去第一个最小的相反数即可,最后求和

code

func largestSumAfterKNegations(nums []int, k int) int {

    for i:=0;i<k;i++{
        sort.Ints(nums)
        nums[0] = -nums[0]
    }
    var sum int
    for _,v:=range nums{
        sum += v
    }
    return sum
}

134. 加油站

在一条环路上有 n 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas 和 cost ,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1 。如果存在解,则 保证 它是 唯一 的。

思路

维护两个值,一个是总sum 一个是当前剩余,如果小于0 从新选择start

code

func canCompleteCircuit(gas []int, cost []int) int {
    curSum:=0
    totalSum:=0
    start:=0

    for i:=0;i<len(gas);i++{
        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
}

你可能感兴趣的:(代码随想录刷题笔记,算法,数据结构)