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

最新推荐文章于 2024-08-22 17:27:46 发布

WDR8965

最新推荐文章于 2024-08-22 17:27:46 发布

阅读量505

点赞数 17

分类专栏：代码随想录算法训练营33期文章标签：算法数据结构 java leetcode

本文链接：https://blog.csdn.net/qq_42973684/article/details/137527271

版权

代码随想录算法训练营33期专栏收录该内容

53 篇文章 2 订阅

订阅专栏

题目与题解

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

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

代码随想录题解：1005.K次取反后最大化的数组和

视频讲解：贪心算法，这不就是常识？还能叫贪心？LeetCode：1005.K次取反后最大化的数组和_哔哩哔哩_bilibili

解题思路：

一个常识性的题，取反后要最大化，那优先负数取反，如果负数已经全部取反且还没达到k次，由于题目中说可以对一个数多次取反，那对绝对值最小的数反复取反一定是最优解。

实际做的时候，一开始想着先从小到大排序，这样方便对前面的负数取反，一边遍历取反一边求取反后的和。如果负数取反结束后，k还有剩余，就对第一个正数反复取反到k结束，但是这样做会出现一个错误，当最大的负数的绝对值比最小的正数小时，得不到最优解。所以就取反到倒数第二个负数，然后再根据最后一个负数和第一个正数的绝对值大小，判断最后取反的数是哪一个，再把k消耗完后对剩余的数求和。写的复杂了一点但也AC了。

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

看完代码随想录之后的想法

随想录的结果更直观，直接根据绝对值进行排序，然后再对最后一个数，即绝对值最小的数反复取反，最后再求和。

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();
		for (int i = 0; i < nums.length && k > 0; i++) {
			if (nums[i] < 0) {
				nums[i] = -nums[i];
				k--;
			}
		}
		if (k % 2 == 1) {
			nums[nums.length - 1] = -nums[nums.length - 1];
		}
		int sum = 0;
		for (int i = 0; i < nums.length; i++) {
			sum += nums[i];
		}
		return sum;
    }
}

遇到的困难

自己的写法也是修修补补才写到最后能ac，但是非常难理解。

随想录的写法很直观，但是绝对值比较的函数写不来，一开始查了以后想重写compare函数，直接用arrays.sort做，但是不知道为什么文心一言给的结果一片红的报错，最后只能放弃了，用stream方法写，这一块不熟。

134. 加油站

题目链接：134. 加油站

代码随想录题解：134. 加油站

视频讲解：贪心算法，得这么加油才能跑完全程！LeetCode ：134.加油站_哔哩哔哩_bilibili

解题思路：

暴力法比较简单，直接计算gas - cost后得到的每次跑完一站剩余油量left，然后以每个加油站为起点，计算遍历一圈后剩余油量之和，如果中途剩余油量小于0，就继续从下一个加油站开始从头遍历，如果最后能跑完一圈，就返回这个起点所在位置。

class Solution {
    public int canCompleteCircuit(int[] gas, int[] cost) {
		if (gas.length == 1) return gas[0] - cost[0] >= 0 ? 0 : -1;
		int number = cost.length;
		int[] left = new int[number];
		for (int i = 0; i < number; i++) {
			left[i] = gas[i]-cost[i];
		}
		for (int i = 0; i < number; i++) {
			if (left[i] > 0) {
				int sum = 0;
				int j = 0;
				for (; j < number; j++) {
					sum += left[(i+j)%number];
					if (sum < 0) break;
				}
				if (j == number) return i;
			}
		}
		return -1;
    }
}

看完代码随想录之后的想法

方法二好理解一点，如果总的剩余油量大于等于0，一定是可以跑一圈的，所以只要遍历一遍，一边计算总剩余油量，一边计算当前剩余油量。如果当前剩余油量小于0了，起点就更新为i+1.

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];
            if (curSum < 0) {
                index = (i + 1) % gas.length ; 
                curSum = 0;
            }
        }
        if (totalSum < 0) return -1;
        return index;
    }
}

遇到的困难

暴力法一开始写不对还超时了，随想录的答案比较巧妙，很难想。

135. 分发糖果

题目链接：135. 分发糖果

代码随想录题解：135. 分发糖果

视频讲解：贪心算法，两者兼顾很容易顾此失彼！LeetCode：135.分发糖果_哔哩哔哩_bilibili

解题思路：

一开始想着先找到数组中最小数字所在的位置，糖果数为1，然后从这个位置分别往两头遍历累加，如果碰到比前一个大的，当前糖果就比前一个糖果多一个，否则就减一个。

但是这样做会出现问题，比如1,3,2,2,1这个序列，最小的在第一个时，第二个糖果数为2，第三个为1，第四个为1，但第五个不能更小了，逻辑有误。

看完代码随想录之后的想法

随想录的结果先从左到右遍历，先记录右边孩子大于左边孩子评分时，当前孩子需要的糖果数量加一，再反向从右到左遍历，根据第一轮记录的结果和左边孩子大于右边孩子评分比较，更新糖果数量，保证左边孩子的糖果数量符合要求。

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