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

文章目录

• 前言
• 一、1005.K次取反后最大化的数组和
• 二、134. 加油站
• 三、135. 分发糖果
• 总结

---

前言

贪心

---

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

• IntStream.of(nums)将int数组转换成一个IntStream类型的流。
• 通过boxed()方法将IntStream流中的每个元素都转换成一个对应的Integer对象，从而将流转换成一个Stream类型的流。
• 通过sorted((o1,o2)->Math.abs(o2)-Math.abs(o1))方法对Stream类型的流进行排序。排序使用的比较器是一个Lambda表达式(o1,o2)->Math.abs(o2)-Math.abs(o1)，该表达式根据o1和o2的绝对值之差来进行比较。因为要按照绝对值从大到小排序，所以o2的绝对值大于o1的绝对值时，返回负值，否则返回正值。
• 通过mapToInt(Integer::intValue)将Stream类型的流转换成一个intStream类型的流。通过toArray()方法将IntStream类型的流转换成一个int类型的数组。

本题思路其实比较好想了，如何可以让数组和最大呢？

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

局部最优可以推出全局最优。

那么如果将负数都转变为正数了，K依然大于0，此时的问题是一个有序正整数序列，如何转变K次正负，让 数组和 达到最大。

那么又是一个贪心：局部最优：只找数值最小的正整数进行反转，当前数值和可以达到最大（例如正整数数组{5, 3, 1}，反转1 得到-1 比 反转5得到的-5 大多了），全局最优：整个 数组和 达到最大。

虽然这道题目大家做的时候，可能都不会去想什么贪心算法，一鼓作气，就AC了。

我这里其实是为了给大家展现出来 经常被大家忽略的贪心思路，这么一道简单题，就用了两次贪心！

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

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

二、134. 加油站

可以换一个思路，首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈，说明 各个站点的加油站 剩油量rest[i]相加一定是大于等于零的。

每个加油站的剩余量rest[i]为gas[i] - cost[i]。

i从0开始累加rest[i]，和记为curSum，一旦curSum小于零，说明[0, i]区间都不能作为起始位置，因为这个区间选择任何一个位置作为起点，到i这里都会断油，那么起始位置从i+1算起，再从0计算curSum。

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

需要注意的是，我们只需要获取index，也就可以成功开始的位置，所以totalSum并不需要在重新定位后更新，只需判断是否大于0即可。

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int cur = 0;
        int total = 0;
        int index = 0;

        for(int i = 0;i<gas.length;i++){
            cur += gas[i] - cost[i];
            total += gas[i] - cost[i];
            if(cur<0){
                index = i+1;
                cur = 0;
            }
        }
        if(total < 0) return -1;
        return index;
    }
}

三、135. 分发糖果

这道题目一定是要确定一边之后，再确定另一边，例如比较每一个孩子的左边，然后再比较右边，如果两边一起考虑一定会顾此失彼。

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

此时局部最优：只要右边评分比左边大，右边的孩子就多一个糖果，全局最优：相邻的孩子中，评分高的右孩子获得比左边孩子更多的糖果

局部最优可以推出全局最优。

再确定左孩子大于右孩子的情况（从后向前遍历）

遍历顺序这里有同学可能会有疑问，为什么不能从前向后遍历呢？

因为 rating[5]与rating[4]的比较 要利用上 rating[5]与rating[6]的比较结果，所以 要从后向前遍历。

class Solution {
    public int candy(int[] ratings) {
        int len = ratings.length;
        int[] candyVec =new int[len];
        candyVec[0] = 1;
        for(int  i = 1;i<len;i++){
            candyVec[i] = (ratings[i] > ratings[i-1]) ? candyVec[i-1]+1 : 1;
        }

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

---

总结

贪心。