【代码随想录】算法训练计划34

贪心

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

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

思路：

• 负数变成正数，很有意思哈，学一下go的sort.Slice

func largestSumAfterKNegations(nums []int, k int) int {
    // 代码一刷
    sort.Slice(nums, func(i,j int) bool {
        return math.Abs(float64(nums[i])) > math.Abs(float64(nums[j]))
    })
    for i:=0; i<len(nums); i++ {
        if k > 0 && nums[i] < 0 {
            k--
            nums[i] = -nums[i]
        }
    }
    if k%2 == 1 {
        nums[len(nums)-1] = -nums[len(nums)-1]
    }
    res := 0
    for i:=0; i<len(nums); i++ {
        res += nums[i]
    }
    return res
}

2、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

思路：

• 很巧妙的一个total，使得看起来像循环一圈能完成了

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
}

3、135. 分发糖果

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

思路：

• 至少为每个人一个糖果
• 两边分别算，取最大值

func candy(ratings []int) int {
    // 贪心
    res1 := make([]int, len(ratings))
    for i:=0; i<len(ratings);i++ {
        res1[i] = 1
    }
    for i:= 1; i<len(ratings); i++ {
        // 右边比左边大的情况，从左开始
        if ratings[i] > ratings[i-1] {
            res1[i] = res1[i-1]+1
        }
        //  else { 不需要减一，别变成了0
        //     res1[i] = res1[i]-1
        // }
    }
    for i:=len(res1)-1; i>0; i-- {
        // 左边比右边大， 从右开始
        if ratings[i-1] > ratings[i] {
            res1[i-1] = max(res1[i-1], res1[i]+1)
        }
    }
    ans := 0
    for i:=0; i<len(res1); i++ {
        ans += res1[i]
    }
    return ans
}
func max(a,b int) int {if a>b {return a}; return b}