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

目录

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

134 加油站

135 分发糖果

---

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

将nums中元素按照绝对值从小到大进行排序。从后向前遍历nums，如果该元素为负数并且k大于0，则将该元素变成正数，并且k--。判断k是否是奇数，如果是则将数组中最小的元素即nums[0]变成-nums[0]，最后返回nums的和。

class Solution {
    public int largestSumAfterKNegations(int[] nums, int k) {
        //按照绝对值大小从小到到排序
        nums = IntStream.of(nums).boxed().sorted((o1,o2)->Math.abs(o1) - Math.abs(o2)).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();
    }
}

时间复杂度O(nlogn)

空间复杂度O(1)

134 加油站

设置变量totalSum和curSum累加gas与cost的差值，idx来记录出发点的下标。

在累加过程中如果curSum小于0说明[0,i]区间中任意一个下标作为出发点都不能成功绕环路行驶一周，则将curSum置为0，并且将idx赋值为i + 1，继续进行累加。累加结束后如果totalSum小于0说明不存在绕环路行驶一周的情况，此时返回-1，否则说明存在可以行驶一周的情况，返回idx。

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

时间复杂度O(n)

空间复杂度O(1)

135 分发糖果

设置数组candies存储每个孩子需要的糖果数。

先从下标1开始遍历到数组的末尾，判断如果当前ratings值比前一个大则赋值为candies[i - 1] + 1否则赋值为1。

再从下标ratings.length - 2开始遍历到下标0，如果当前ratings值比后一个大则将candies取candies[i + 1] + 1与candies[i]中的最大值。

最后返回candies中值的总和。

class Solution {
    public int candy(int[] ratings) {
        int candies[] = new int[ratings.length];
        candies[0] = 1;
        for(int i = 1;i < ratings.length;i++){
            candies[i] = ratings[i] > ratings[i - 1] ? candies[i - 1] + 1  : 1;
        }
        for(int i = ratings.length - 2;i >= 0;i--){
            if(ratings[i] > ratings[i + 1])candies[i] = Math.max(candies[i + 1] + 1,candies[i]);   
        }
        int res = 0;
        for(int i = 0;i < candies.length;i++)res += candies[i];
        return res;
    }
}

时间复杂度O(n)

空间复杂度O(n)