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

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

题目链接：https://leetcode.cn/problems/maximize-sum-of-array-after-k-negations/
文档讲解：https://programmercarl.com/1005.K%E6%AC%A1%E5%8F%96%E5%8F%8D%E5%90%8E%E6%9C%80%E5%A4%A7%E5%8C%96%…
视频讲解：https://www.bilibili.com/video/BV138411G7LY

思路

• 第一次贪心：优先取绝对值大的负数进行取反。
• 第二次贪心：当数组中没有负数了，取最小的正数进行取反，消耗掉剩余取反次数。

代码

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        Arrays.sort(nums);
        int sum = 0;
        for (int i = 0; i < nums.length; i++) {
            if(nums[i] < 0 && k > 0) {
                nums[i] = -nums[i];
                k--;
            } else break;
        }
        if (k > 0) {
            Arrays.sort(nums);
            if ((k % 2) != 0)  nums[0] = -nums[0];
        }
        for (int i = 0; i < nums.length; i++) sum += nums[i];
        return sum;
    }
}

分析：时间复杂度：O(nlogn)，空间复杂度：O(1)。
Arrays.sort(nums);的时间复杂度是O(nlogn)。

134. 加油站

题目链接：https://leetcode.cn/problems/gas-station/
文档讲解：https://programmercarl.com/0134.%E5%8A%A0%E6%B2%B9%E7%AB%99.html
视频讲解：https://www.bilibili.com/video/BV1jA411r7WX

思路

• 计算加油和消耗的差值。从第一个站点开始计算油量，如果出现负数，说明从前面的站点开始走不能走完全程，于是将开始的站点startIndex设为下一个，重新计算差值的和。
• 同时也计算全程的油量差，如果全程的油量差是负数（加油量少于消耗量），则返回-1。否则返回startIndex。

代码

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

分析：时间复杂度：O(n)，空间复杂度：O(1)。

135. 分发糖果

题目链接：https://leetcode.cn/problems/candy/
文档讲解：https://programmercarl.com/0135.%E5%88%86%E5%8F%91%E7%B3%96%E6%9E%9C.html
视频讲解：https://www.bilibili.com/video/BV1ev4y1r7wN

思路

• 情况一：右孩比左孩得分高。需要从前往后遍历。

for (int i = 1; i < ratings.length; i++) {
            if (ratings[i] > ratings[i - 1]) candy[i] = candy[i - 1] + 1;
        }

• 情况二：左孩比右孩得分高。需要在情况一的基础上从后往前遍历。取两种情况下糖果多的情况。

for (int i = ratings.length - 2; i >= 0; i--) {
            if (ratings[i] > ratings[i + 1]) candy[i] = Math.max(candy[i + 1] + 1, candy[i]); // 取最大值保证中间的糖果数比左右都大
        }

代码

class Solution {
    public int candy(int[] ratings) {
        int[] candy = new int[ratings.length];
        Arrays.fill(candy, 1);
        // 右孩比左孩大
        for (int i = 1; i < ratings.length; i++) {
            if (ratings[i] > ratings[i - 1]) candy[i] = candy[i - 1] + 1;
        }
        // 左孩比右孩大
        for (int i = ratings.length - 2; i >= 0; i--) {
            if (ratings[i] > ratings[i + 1]) candy[i] = Math.max(candy[i + 1] + 1, candy[i]);
        }
        int sum = 0;
        for (int i = 0; i < ratings.length; i++) sum += candy[i];
        return sum;
    }
}

分析：时间复杂度：O(n)，空间复杂度：O(n)。