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

学习目标

• 锻炼贪心算法相关题目

---

学习内容

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

题目链接:1005.K次取反后最大化的数组和

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

示例 1：
输入：nums = [4,2,3], k = 1
输出：5
解释：选择下标 1 ，nums 变为 [4,-2,3] 。

示例 2：
输入：nums = [3,-1,0,2], k = 3
输出：6
解释：选择下标 (1, 2, 2) ，nums 变为 [3,1,0,2] 。

示例 3：
输入：nums = [2,-3,-1,5,-4], k = 2
输出：13
解释：选择下标 (1, 4) ，nums 变为 [2,3,-1,5,4] 。

提示：
1 <= nums.length <= 104
-100 <= nums[i] <= 100
1 <= k <= 104

思路：按绝对值大小从大到小排序, 再更改其中的负值. 若 遍历数组结束后 K > 0, 则让 K 反复将绝对值最小的数更改.

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

解决方案：

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        nums = IntStream.of(nums)
                .boxed()
                .sorted((a,b)->Math.abs(b) - Math.abs(a))
                .mapToInt(Integer::intValue).toArray();

        for(int i = 0; i < nums.length; i++) {
            if(nums[i] < 0 && k > 0) {
                nums[i] = - nums[i];
                k --;
            }
        }
        if(k % 2 == 1) nums[nums.length - 1] = -  nums[nums.length - 1];

        return Arrays.stream(nums).sum();
    }
}

小结：贪心的思路，局部最优：让绝对值大的负数变为正数，当前数值达到最大，整体最优：整个数组和达到最大。

134. 加油站(Middle)

题目链接:134. 加油站

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

示例 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 可为起始索引。

示例 2:
输入: gas = [2,3,4], cost = [3,4,3]
输出: -1
解释:
你不能从 0 号或 1 号加油站出发，因为没有足够的汽油可以让你行驶到下一个加油站。
我们从 2 号加油站出发，可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
开往 0 号加油站，此时油箱有 4 - 3 + 2 = 3 升汽油
开往 1 号加油站，此时油箱有 3 - 3 + 3 = 3 升汽油
你无法返回 2 号加油站，因为返程需要消耗 4 升汽油，但是你的油箱只有 3 升汽油。
因此，无论怎样，你都不可能绕环路行驶一周。

提示:
gas.length == n
cost.length == n
1 <= n <= 105
0 <= gas[i], cost[i] <= 104

思路：无
时间复杂度：O(n) 空间复杂度：O(1)

解决方案：

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int totalSum = 0;
        int currSum = 0;
        int start = 0;

        for(int i = 0; i < gas.length; i++) {
            currSum += (gas[i] - cost[i]);
            totalSum += (gas[i] - cost[i]);
            
            if(currSum < 0) {
                currSum = 0;
                start = i + 1;
            }
        }

        if(totalSum < 0) return -1;
        return start;
    }
}

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

135. 分发糖果(Middle)

题目链接:135. 分发糖果

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

示例 1：
输入：ratings = [1,0,2]
输出：5
解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2：
输入：ratings = [1,2,2]
输出：4
解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。

提示：
n == ratings.length
1 <= n <= 2 * 104
0 <= ratings[i] <= 2 * 104

思路：先从左边开始贪心, 若 ratings[i] <= ratings[i-1], 则 ans[i] 可以只给 1 颗糖, 否则给 ans[i-1] + 1 颗;
再从右边开始贪心, 若 ratings[i] < ratings[i+1], 则 ans[i] 的数量在 ratings[i+1] + 1 和 ans[i] 中取最大值, 保证比左右两边都大.
时间复杂度：O(n) 空间复杂度：O(n)

解决方案：

class Solution {
    public int candy(int[] ratings) {
        if(ratings.length == 1) return 1;
        int[] ans = new int[ratings.length];
        ans[0] = 1;

        for(int i = 1; i < ratings.length; i++) {
            ans[i] = ratings[i] <= ratings[i-1] ? 1 : ans[i - 1] + 1;
        }

        for(int i = ratings.length - 2; i >= 0; i--) {
            if(ratings[i] > ratings[i+1]) {
                ans[i] = Math.max(ans[i], ans[i+1] + 1);
            }
        }

        return Arrays.stream(ans).sum();
    }
}

小结：这在leetcode上是一道困难的题目，其难点就在于贪心的策略，如果在考虑局部的时候想两边兼顾，就会顾此失彼。

Carl 哥的题解提出可以使用两次贪心策略:

一次是从左到右遍历，只比较右边孩子评分比左边大的情况。
一次是从右到左遍历，只比较左边孩子评分比右边大的情况。

这样从局部最优推出了全局最优，即：相邻的孩子中，评分高的孩子获得更多的糖果。

---

总结

贪心算法在于找出局部解和全局解, 找局部最优解这部分仍然比较匮乏, 需要多练习. 而进入的 LC135 则提供了一个新的思路.