代码随想录day29：贪心part3，stream处理数组的思路

最新推荐文章于 2024-08-29 16:57:08 发布

O_ra_Cle

最新推荐文章于 2024-08-29 16:57:08 发布

阅读量264

点赞数 3

分类专栏：代码随想录文章标签：算法数据结构 java

本文链接：https://blog.csdn.net/O_ra_Cle/article/details/136408165

版权

代码随想录专栏收录该内容

46 篇文章 0 订阅

订阅专栏

本文介绍了三种编程题目，分别基于贪心策略解决：1.K次取反后最大化数组和；2.加油站问题，找到能完成一圈的最大行驶距离；3.分发糖果问题，计算满足条件的最大糖果分配。通过Java实现了解题思路。

摘要由CSDN通过智能技术生成

文章目录

- day29：贪心part3

day29：贪心part3

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

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        nums = IntStream.of(nums)
                        .boxed()
                        .sorted((a, b) -> Math.abs(a) - Math.abs(b))
                        .mapToInt(Integer::intValue).toArray();
        for (int i = nums.length - 1; i >= 0; i--) {
            if (nums[i] < 0 && k > 0) {
                nums[i] = -nums[i];
                k--;
            }
        }
        if (k % 2 == 1) nums[0] = -nums[0];
        return Arrays.stream(nums).sum();
    }
}

134.加油站

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
        int curSum = 0;
        int totalSum = 0;
        int start = 0;
        for (int i = 0; i < gas.length; i++) {
            curSum += gas[i] - cost[i];
            totalSum += gas[i] - cost[i];
            if (curSum < 0) {
                start = i + 1;
                curSum = 0;
            }
        }
        if (totalSum < 0) return -1;
        else return start;
    }
}

135.分发糖果

class Solution {
    public int candy(int[] ratings) {
        // 两次贪心：
        // 从前往后遍历处理右边孩子大于左边孩子的情况
        // 从后往前遍历处理左大于右的情况
        int[] candy = new int[ratings.length];
        candy[0] = 1;
        for (int i = 1; i < ratings.length; i++) {
            if (ratings[i] > ratings[i - 1])
                candy[i] = candy[i - 1] + 1;
            else candy[i] = 1;
        }
        for (int i = ratings.length - 2; i >= 0; i--) {
            if (ratings[i] > ratings[i + 1])
                candy[i] = Math.max(candy[i], candy[i + 1] + 1);
        }
        return Arrays.stream(candy).sum();
    }
}