算法学习day34
1. 力扣1005.K次取反后最大化的数组和
1.1 分析
首先,函数定义优先队列pq,并使用循环将 nums 中的每个元素插入到pq中,并累加它们的和 sum。在循环结束后,pq中的所有元素都已经按照从小到大的顺序排列了,sum 存储了数组中所有元素的和。
然后,函数开始进行 k 次修改。在每次循环中,函数从pq中取出最小的元素 minNum,将其从堆中移除,并将其相反数 -minNum 插入到堆中。同时,函数将 sum 减去 minNum,并将其加上 -minNum。这一过程相当于将 minNum 替换为 -minNum,并更新 sum。
最后,函数返回更新后的 sum,即为修改 k 次后数组的可能最大和。
1.2 代码
class Solution {
public:
int largestSumAfterKNegations(vector<int>& nums, int k) {
priority_queue<int,vector<int>,greater<int>> pq;
int sum = 0;
for(int num: nums){
pq.push(num); // 将数组的数值插入
sum += num; // 统计数组中的数的和
}
while(k--){
int minNum = pq.top(); // 取出最小的数
pq.pop(); // 将最小的数弹出
sum -= minNum; // 减去最小的数
pq.push(-minNum); // 将其相反数插入
sum += -minNum;
}
return sum;
}
};
2. 力扣134. 加油站
2.1 分析
首先我们需要判断是否有解,即整个加油站的总油量是否大于等于总消耗油量,如果总油量小于总消耗油量,那么一定无解。
假设从i号加油站出发,一直到k号加油站都能正常行驶,但在k+1号加油站时油量不够,导致无法到达k+2号加油站,那么从任何一个i到k之间的加油站出发都无法到达k+1号加油站,因此这些加油站都不可能成为起点,所以直接从k+1号加油站开始考虑即可。
2.2 代码
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){ // 当前油量小于0,说明无法到达下一个加油站
start = i + 1; // 更新起始位置
curSum = 0; // 将当前油量归0,重新开始计算当前油量
}
}
if(totalSum < 0) return -1; // 总油量小于总消耗量,无法回到起点
return start; // 返回起点位置
}
};
3. 力扣135. 分发糖果
3.1 分析
先确定一边,再确定另外一边。首先比较每个孩子的左边,再比较右边。先确定右边评分大于左边的情况(从前向后遍历)
此时局部最优:只要右边评分比左边大,右边孩子多一个糖果。
此时全局最优:在相邻孩子中,评分高的右孩子比左孩子得更多的糖果。
for(int i = 1; i < ratings.size(); i++){
if(rating[i] > ratings[i - 1]) candyVec[i] = candyVec[i - 1] + 1;
}
在确定左孩子大于右孩子的情况(从后向前遍历)
for(int i = ratings.size() - 2 ; i >=0; i--){
if(ratings[i] > ratings[i+1]){
candyVec[i] = max(candyVec[i],candyVec[i+1] + 1);
}
}
3.2 代码
class Solution {
public:
int candy(vector<int>& ratings) {
vector<int> candyVec(ratings.size(),1); // 初始化每个孩子的糖果数为1
// 从前向后遍历
for(int i = 1 ; i < ratings.size() ; i++){
if( ratings[i] > ratings[i-1]) candyVec[i] = candyVec[i-1] + 1; // 如果当前孩子的评分高于前一个孩子,就将糖果数加1
}
// 从后向前遍历
for(int i = ratings.size() -2 ; i >= 0 ; i--){
if(ratings[i] > ratings[i + 1]){ // 如果当前孩子的评分高于后一个孩子
candyVec[i] = max(candyVec[i],candyVec[i+1] + 1); // 将当前孩子的糖果数与后一个孩子的糖果数加1后的数取最大值
}
}
int result = 0;
for(int i = 0 ; i< candyVec.size(); i++) result += candyVec[i]; // 统计所有孩子的糖果数之和
return result;
}
};