心得:for循环适合模拟从头到尾的遍历,而while循环适合模拟环形遍历,要善于使用while!
第一题,贪心两次,先把最小的负数取反,若还有剩余次数,就再贪心把最小的正数取反。
第二题,加油站,用剩余油量计算,若剩余油量小于零,则更新起始位置为当前位置的后一位。因为前面的任何一个地方都不可能有合适的起始位置。
第三题,贪心的局部最优体现在先只局部判断右边的,再局部判断左边的(同时取这两个的最大)
第一题,K次取反后最大化的数组和 https://leetcode.cn/problems/maximize-sum-of-array-after-k-negations/
给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后,返回数组 可能的最大和 。
这道题可以谈心两次,先把最小的负数取反,若还有剩余次数,就再贪心把最小的正数取反。
class Solution {
public:
int largestSumAfterKNegations(vector<int>& nums, int k) {
int res = 0;
int minVal = INT_MAX;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size(); i++){
if(nums[i] < 0 && k > 0){
nums[i] = -nums[i];
k--;
}
res += nums[i];
minVal = min(nums[i], minVal);
}
if(k % 2 == 1){
res = res - 2 * minVal;
}
return res;
}
};
第二题、加油站 LeetCode134 https://leetcode.cn/problems/gas-station/
在一条环路上有 n 个加油站,其中第 i 个加油站有汽油 gas[i] 升。
你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发,开始时油箱为空。
给定两个整数数组 gas 和 cost ,如果你可以按顺序绕环路行驶一周,则返回出发时加油站的编号,否则返回 -1 。如果存在解,则 保证 它是 唯一 的。
当然可以直接暴力遍历整个路径,但是要注意for循环适合模拟从头到尾的遍历,而while循环适合模拟环形遍历,要善于使用while!
而这一题的贪心算法是用了剩余油量(加的油减去消耗的油)。
首先如果总油量减去总消耗大于等于零那么一定可以跑完一圈,说明 各个站点的加油站 剩油量rest[i]相加一定是大于等于零的。
每个加油站的剩余量rest[i]为gas[i] - cost[i]。
i从0开始累加rest[i],和记为curSum,一旦curSum小于零,说明[0, i]区间都不能作为起始位置,因为这个区间选择任何一个位置作为起点,到i这里都会断油,那么起始位置从i+1算起,再从0计算curSum。
如图:
那么为什么一旦[0,i] 区间和为负数,起始位置就可以是i+1呢,i+1后面就不会出现更大的负数?
如果出现更大的负数,就是更新i,那么起始位置又变成新的i+1了。
那有没有可能 [0,i] 区间 选某一个作为起点,累加到 i这里 curSum是不会小于零呢? 如图:
如果 curSum<0 说明 区间和1 + 区间和2 < 0, 那么 假设从上图中的位置开始计数curSum不会小于0的话,就是 区间和2>0。
区间和1 + 区间和2 < 0 同时 区间和2>0,只能说明区间和1 < 0, 那么就会从假设的箭头初就开始从新选择其实位置了。
那么局部最优:当前累加rest[i]的和curSum一旦小于0,起始位置至少要是i+1,因为从i之前开始一定不行。全局最优:找到可以跑一圈的起始位置。
局部最优可以推出全局最优,找不出反例,试试贪心!
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int curSum = 0;
int totalSum = 0;
int start = 0;
for(int i = 0; i < gas.size(); i++){
curSum += gas[i] - cost[i];
totalSum += gas[i] - cost[i];
if(curSum < 0){ // 当前累加rest[i]和 curSum一旦小于0
start = i + 1; // 起始位置更新为i+1
curSum = 0; // curSum从0开始
}
}
if(totalSum < 0) return -1;
return start;
}
};
第三题、分发糖果 LeetCode135 https://leetcode.cn/problems/candy/
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
条件一:每个孩子至少分配到 1 个糖果。
条件二:相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
这道题的要分保证某个元素左右都满足条件二。
如果两边一起考虑一定会顾此失彼。
先确定右边评分大于左边的情况(也就是从前向后遍历)
此时局部最优:只要右边评分比左边大,右边的孩子就多一个糖果,全局最优:相邻的孩子中,评分高的右孩子获得比左边孩子更多的糖果
局部最优可以推出全局最优。
class Solution {
public:
int candy(vector<int>& ratings) {
vector<int> candy(ratings.size(), 1);
//for(int i = 0; i < ratings.size(); i++){
for(int i = 1; i < ratings.size(); i++){
if(ratings[i] > ratings[i - 1]){
candy[i] = candy[i - 1] + 1;
}
}
//for(int i = ratings.size() - 1; i > 0; i--){
for(int i = ratings.size() - 2; i >= 0; i--){
if(ratings[i] > ratings[i + 1]){
//candy[i] = candy[i + 1] + 1
candy[i] = max(candy[i], candy[i + 1] + 1);//从右往左的局部最优并综合全局最优
}
}
int res = 0;
for(int i = 0; i < ratings.size(); i ++){
res += candy[i];
}
return res;
}
};