代码随想录算法训练营第三十三天|LeetCode1005,134,135
1005.K次取反后最大化的数组和
自己的思路:
对数组进行排序
对小于0的进行取反,如果K已经取完了,就进行求和。
如果k没有去完,则对数组重新排序。然后对最小的那个值反复取反。
其实上面两个就算是贪心的思路。
public int largestSumAfterKNegations(int[] nums, int k) {
int sum = 0;
Arrays.sort(nums);
for (int i = 0 ; i<nums.length;i++){
if (nums[i] >= 0){
break;
}
if (k > 0 && nums[i] < 0){
nums[i] = -nums[i];
k--;
continue;
}
if (k > 0 && nums[i] >= 0 ){
break;
}
}
Arrays.sort(nums);
while(k > 0){
nums[0] = - nums[0];
k--;
}
for (int i = 0; i < nums.length;i++){
sum+=nums[i];
}
return sum;
}
134. 加油站
第一想法是暴力法:从第一个节点开始,一个个的试,如果中间没有出现断油的情况,则证明可行。但是,这种方法超过了时间限制。但通过暴力法的尝试,可以明确:for循环更适应顺序遍历,while循环更适应环形遍历
贪心法:
首先明确的一点是:总油量>=总消耗量,那么肯定可以跑完一圈的
设定rest[i]为剩余油量
然后在循环每个油站的时候累加,一旦出现负值,那么从开始计算curSum的位置到出现负值的位置中间都不可能会有起点,因为到负值这必出现断油,此时,从下一位置重新计算curSum。
使用贪心就不用考虑环形遍历了。
public int canCompleteCircuit(int[] gas, int[] cost) {
//记录剩余油量之和
int curSum = 0;
//记录总共的剩余油量和
int totalSum = 0;
//记录起点位置。
int startIndex = 0;
for (int i = 0; i < gas.length; i++){
//更新剩余油量
curSum += gas[i] - cost[i];
totalSum += gas[i] - cost[i];
if (curSum < 0){
curSum = 0;
startIndex = i+1;
}
}
if (totalSum<0) return -1;
return startIndex;
}
135. 分发糖果
这道题目一定是要确定一边之后,再确定另一边,例如比较每一个孩子的左边,然后再比较右边,如果两边一起考虑一定会顾此失彼。
先确定右边评分大于左边的情况(也就是从前向后遍历)
此时局部最优:只要右边评分比左边大,右边的孩子就多一个糖果,全局最优:相邻的孩子中,评分高的右孩子获得比左边孩子更多的糖果
再确定左孩子大于右孩子的情况(从后向前遍历)如果 ratings[i] > ratings[i + 1],此时candyVec[i](第i个小孩的糖果数量)就有两个选择了,一个是candyVec[i + 1] + 1(从右边这个加1得到的糖果数量),一个是candyVec[i](之前比较右孩子大于左孩子得到的糖果数量)。
那么又要贪心了,局部最优:取candyVec[i + 1] + 1 和 candyVec[i] 最大的糖果数量,保证第i个小孩的糖果数量既大于左边的也大于右边的
public int candy(int[] ratings) {
int[] candyVec = new int[ratings.length];
//保证每个小孩都有一个糖
Arrays.fill(candyVec,1);
//先从左往右遍历,确定右边孩子比左边孩子大的多一个糖果。
for(int i = 1; i< ratings.length;i++){
if (ratings[i] > ratings[i-1]){
candyVec[i] = candyVec[i-1]+1;
}
}
//然后从右往左遍历,判断左边孩子比右边孩子多一个糖果,
//这一步必须从右往左遍历,因为这一步需要用到左孩子与左孩子的左孩子的判断结果。
for (int i = ratings.length -2; i>=0;i--){
if (ratings[i] > ratings[i+1]){
//这里要取candyVec[i]和candyVec[i+1]+1的最大值
candyVec[i] = Math.max(candyVec[i], candyVec[i+1]+1);
}
}
int result = 0;
for (int i:candyVec){
result+=i;
}
return result;
}