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

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

1005. K 次取反后最大化的数组和 - 力扣(LeetCode)

代码随想录 (programmercarl.com)

贪心算法,这不就是常识?还能叫贪心?LeetCode:1005.K次取反后最大化的数组和_哔哩哔哩_bilibili

给你一个整数数组 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

 第一次贪心:局部最优:绝对值最大的负数变成正数;全局最优:和最大

第二次贪心:局部最优:数值小的正数变成负数;全局最优:和最大

具体步骤:

1、将数组按照绝对值大小从大到小排列

2、遍历数组

3、遇到负数就将负数变为正数,同时k--

4、负数改完了k仍然不为0、

5、将数组末尾的最小的正数值反复变负数又变正数,直到k=0

class Solution {
    public int largestSumAfterKNegations(int[] nums, int K) {
        // 将数组按照绝对值大小从大到小排序,注意要按照绝对值的大小
        nums = IntStream.of(nums)  // 将原始数组转换为 IntStream
            .boxed()  // 装箱,将基本数据类型转换为包装类型,以便使用 sorted 方法
            .sorted((o1, o2) -> Math.abs(o2) - Math.abs(o1))  // 按绝对值大小排序
            .mapToInt(Integer::intValue).toArray();  // 将排序后的 IntStream 转换为 int 数组
        int len = nums.length;  // 数组长度
        for (int i = 0; i < len; i++) {
            // 从前向后遍历,遇到负数将其变为正数,同时 K--
            if (nums[i] < 0 && K > 0) {
                nums[i] = -nums[i];  // 将负数变为正数
                K--;  // K 减一
            }
        }
        // 如果 K 还大于 0,那么反复转变数值最小的元素,直到 K 用完
        if (K % 2 == 1) nums[len - 1] = -nums[len - 1];  // 如果 K 为奇数,将最后一个元素变为负数
        return Arrays.stream(nums).sum();  // 返回数组元素的和
    }
}

134. 加油站 

134. 加油站 - 力扣(LeetCode)

代码随想录 (programmercarl.com)

贪心算法,得这么加油才能跑完全程!LeetCode :134.加油站_哔哩哔哩_bilibili

在一条环路上有 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

假设有以下情况:

当汽车从下标为0的位置开始走,当走到下标为3的位置时, 剩余的油不够在这里的消耗,所以没有办法走到尾部,此时从下标为4的位置重新开始走,代码如下:

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        // 初始化当前累计油量、总累计油量和起始加油站索引
        int curSum = 0;
        int totalSum = 0;
        int index = 0;
        // 遍历每个加油站
        for (int i = 0; i < gas.length; i++) {
            // 计算当前累计油量和总累计油量
            curSum += gas[i] - cost[i];
            totalSum += gas[i] - cost[i];
            // 如果当前累计油量为负,则说明无法从当前加油站出发,更新起始加油站索引,并将当前累计油量重置为0
            if (curSum < 0) {
                index = (i + 1);
                curSum = 0;
            }
        }
        // 如果总累计油量为负,则说明无法绕行一圈,返回-1;否则返回起始加油站索引
        if (totalSum < 0) return -1;
        return index;
    }
}

135. 分发糖果 

135. 分发糖果 - 力扣(LeetCode)

代码随想录 (programmercarl.com)

贪心算法,两者兼顾很容易顾此失彼!LeetCode:135.分发糖果_哔哩哔哩_bilibili

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

精髓在于不要同时比较左右

先从前向后遍历确定右边评分大于左边的情况,再从后往前遍历确定左边评分大于右边的情况。

从前向后遍历确定右边评分大于左边的情况:

局部最优:右边评分比左边大,右边的孩子就多一颗糖果;全局最优:相邻孩子中,评分高的右孩子比左孩子获得更多的糖果。

 for (int i = 1; i < len; i++) {
            candyVec[i] = (ratings[i] > ratings[i - 1]) ? candyVec[i - 1] + 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);
            }
        }

综合代码:

class Solution {
    /**
     * 分两个阶段
     * 1、起点下标1 从左往右,只要 右边 比 左边 大,右边的糖果=左边 + 1
     * 2、起点下标 ratings.length - 2 从右往左, 只要左边 比 右边 大,此时 左边的糖果应该 取本身的糖果数(符合比它左边大) 和 右边糖果数 + 1 二者的最大值,这样才符合 它比它左边的大,也比它右边大
     */
    public int candy(int[] ratings) {
        int len = ratings.length;  // 计算评分数组的长度
        int[] candyVec = new int[len];  // 创建一个与评分数组等长的数组,用于存储每个孩子分到的糖果数量
        candyVec[0] = 1;  // 第一个孩子至少分到一个糖果

        // 第一次遍历,从左往右分糖果,只要右边的孩子评分比左边的大,右边的孩子糖果数就比左边多1
        for (int i = 1; i < len; i++) {
            candyVec[i] = (ratings[i] > ratings[i - 1]) ? candyVec[i - 1] + 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;  // 返回总共需要的糖果数
    }
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值