代码随想录算法训练营第三十四天|1005.K次取反后最大化的数组和、134.加油站、135.分发糖果
1005.K次取反后最大化的数组和
给你一个整数数组 nums
和一个整数 k
,按以下方法修改该数组:
- 选择某个下标
i
并将nums[i]
替换为-nums[i]
。
重复这个过程恰好 k
次。可以多次选择同一个下标 i
。
以这种方式修改数组后,返回数组 可能的最大和 。
示例 1:
输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3] 。
题解:需要注意的点是:该怎么排序呢?很明心爱先将负数取反,所有负数取反后K还有剩余的话,对数组里最小的那个值取反,怎么排序才能找到这个最小的数呢?
按绝对值排序:单独写一个排序算法。
这个方法没实现出来,还是老老实实重新拍一次序吧。
代码:
class Solution {
public int largestSumAfterKNegations(int[] nums, int k) {
Arrays.sort(nums);
int res=0;
for(int i=0;i<nums.length;i++){
//如果遍历到的值是负数
if(nums[i]<0 && k>0){
nums[i]=-nums[i];
k--;
}
res+=nums[i];
}
//从小到大排序
Arrays.sort(nums);
res-=nums[0]*(k%2==1?2:0);
return res;
}
}
134.加油站
在一条环路上有 n
个加油站,其中第 i
个加油站有汽油 gas[i]
升。
你有一辆油箱容量无限的的汽车,从第 i
个加油站开往第 i+1
个加油站需要消耗汽油 cost[i]
升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas
和 cost
,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1
。如果存在解,则 保证 它是 唯一 的。
示例 1:
输入: gas = [1,2,3,4,5], cost = [3,4,5,1,2]
输出: 3
解释:
从 3 号加油站(索引为 3 处)出发,可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站,此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站,此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站,此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站,此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站,你需要消耗 5 升汽油,正好足够你返回到 3 号加油站。
因此,3 可为起始索引。
题解:计算到达每一个加油站的补充和消耗的差值,从头开始累加这个差值,如果小于0,说明油不够到达下一个站点,那么将遍历的下一个节点作为开始值。
代码:
class Solution {
public int canCompleteCircuit(int[] gas, int[] cost) {
int start=0,totalsum=0,currsum=0;
for(int i=0;i<gas.length;i++){
currsum+=(gas[i]-cost[i]);
totalsum+=(gas[i]-cost[i]);
if(currsum<0){
//记录开始位置
start=i+1;
//差值和重置为0
currsum=0;
}
}
if(totalsum<0) return -1;
return start;
}
}
135.分发糖果
n
个孩子站成一排。给你一个整数数组 ratings
表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
- 每个孩子至少分配到
1
个糖果。 - 相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
示例 1:
输入:ratings = [1,0,2]
输出:5
解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
题解:分别考虑两次。一次考虑右边孩子比左边孩子大的情况,一次考虑左边比右边大的情况,然后取两次的最大值就是孩子应该拥有的糖果数。
两次遍历,一次从前向后,一次从后向前。
代码:
class Solution {
public int candy(int[] ratings) {
int[] can=new int[ratings.length];
//初始化,所有小孩都有一颗糖果
Arrays.fill(can,1);
int ans=0;
//右边比左边大
for(int i=1;i<ratings.length;i++){
if(ratings[i]>ratings[i-1]){
can[i]=can[i-1]+1;
}
}
//左边比右边大
for(int i=ratings.length-2;i>=0;i--){
if(ratings[i]>ratings[i+1]){
can[i]=Math.max(can[i+1]+1,can[i]);
}
ans+=can[i];
}
//for循环没加数组最后一个值
return ans+=can[can.length-1];
}
}