题目与题解
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. 加油站
解题思路:
暴力法比较简单,直接计算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. 分发糖果
解题思路:
一开始想着先找到数组中最小数字所在的位置,糖果数为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;
}
}
遇到的困难
遇到不对就不知道怎么做了,想不到可以两头更新。
今日收获
今天又学了新的贪心算法,感觉还是想不到,要记的东西越来越多了TAT。