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

78 阅读2分钟

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

题目链接:1005. K 次取反后最大化的数组和

思路: 本题贪心逻辑:碰到负数直接取反,难点在于,如果负数的个数要小于k,那么剩下几次取反应该如何操作。如果剩下的次数为偶数次,那么相当于0次(偶数次取反一个数还是本身)。如果剩下的次数为奇数次,那么相当于1次,此时需要找到最小的数,将它取反。

先把数组按照绝对值大小从大到小排列, 碰到负数直接取反。如果剩下的次数为偶数次,直接返回数组和,如果剩下的次数为奇数次,将数组最后一位数取反,返回数组和。

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++) {
	    //从前向后遍历,遇到负数将其变为正数,同时K--
	        if (nums[i] < 0 && k > 0) {
	    	    nums[i] = -nums[i];
	    	    k--;
	        }
	    }
	    // 如果K还大于0,那么反复转变数值最小的元素,将K用完

	    if (k % 2 == 1) nums[len - 1] = -nums[len - 1];
	    return Arrays.stream(nums).sum();
    }
}

134. 加油站

题目链接:134. 加油站

思路: 暴力的解法是,对每一个加油站当作起点,不断累加gas和cost的差,如果res一直大于等于0,遍历完之后,返回这个起点。Java会超时。

贪心的思路是:i从0开始累加rest[i],和记为curSum,一旦curSum小于零,说明[0, i]区间都不能作为起始位置,起始位置从i+1算起,再从0计算curSum。

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

135. 分发糖果

题目链接:135. 分发糖果

思路: 先从左至右遍历学生成绩 ratings,再从右至左遍历,这样可以保证同时满足左规则和右规则。

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