代码随想录算法训练营day34 | 1005.K次取反后最大化的数组和 ,134. 加油站,135. 分发糖果
- 1005.K次取反后最大化的数组和
- 解法一:两次正常排序
- 解法二:一次排序
- 134. 加油站
- 135. 分发糖果
1005.K次取反后最大化的数组和
教程视频:https://www.bilibili.com/video/BV138411G7LY/?spm_id_from=333.788&vd_source=ddffd51aa532d23e6feac69924e20891
解法一:两次正常排序
第一次取绝对值最大负数,第二次去绝对值最小非负数。
class Solution {
public int largestSumAfterKNegations(int[] nums, int k) {
Arrays.sort(nums);
int count = 0;
for(int i=0;i<nums.length;i++){
if(nums[i]<0 && k>0){
k--;
nums[i]*=-1;
}
}
Arrays.sort(nums);
if(k>0 && k%2==1){
nums[0]*=-1;
}
int sum=0;
for(int i=0;i<nums.length;i++){
sum+=nums[i];
}
return sum;
}
}
解法二:一次排序
直接按照绝对值排序
利用了java8的新特性:stream
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. 加油站
教程视频:https://www.bilibili.com/video/BV1jA411r7WX/?spm_id_from=333.788&vd_source=ddffd51aa532d23e6feac69924e20891
【注意】想清楚局部最优解,不要陷入数组循环的误区。
局部最优: 到第 i 天累加剩余油量curSum一旦小于0,起始位置至少要是i+1,因为从i之前开始一定不行。
全局最优: 找到可以跑一圈的起始位置。
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);
curSum = 0;
}
}
if (totalSum < 0) return -1;
return index;
}
}
135. 分发糖果
教程视频:https://www.bilibili.com/video/BV1ev4y1r7wN/?spm_id_from=333.788&vd_source=ddffd51aa532d23e6feac69924e20891
解题关键:先确定一个方向,再确定另一个方向,不要同时考虑两边,否则会顾此失彼。
(按照题意,等于的情况和小于一样处理)
因此位置 i 共有两种情况需要增大:
1、比前一个数大,即ratings[i]>ratings[i-1],此时小的在左边,这种情况从前向后遍历
2、比后一个数大,即ratings[i]>ratings[i+1],此时小的在右边,这种情况从后向前遍历
class Solution {
public int candy(int[] ratings) {
int[] candy = new int[ratings.length];
candy[0]=1;
//右大于左,即ratings[i]>ratings[i-1],小的在左边,这种情况从前向后遍历
for(int i=1;i<candy.length;i++){
candy[i] = (ratings[i-1]<ratings[i]) ? candy[i-1]+1 : 1;
}
//左大于右,即ratings[i]>ratings[i+1],小的在右边,这种情况从后向前遍历
for(int i=candy.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;
}
}