代码随想录算法训练营第34天|1005.K次取反后最大化的数组和,134. 加油站,135. 分发糖果
1005.K次取反后最大化的数组和
题目链接:1005.K次取反后最大化的数组和,难度:简单
【实现代码】
class Solution {
public:
static bool cmp(int a, int b) {
return abs(a) > abs(b);
}
int largestSumAfterKNegations(vector<int>& nums, int k) {
sort(nums.begin(), nums.end(), cmp);
for (int i = 0; i < nums.size(); i++) {
if (nums[i] < 0 && k > 0) {
nums[i] = -nums[i];
k--;
}
}
if (k % 2 == 1) {
nums[nums.size() - 1] *= -1;
k--;
}
int result = 0;
for(int i = 0; i < nums.size(); i++) {
result += nums[i];
}
return result;
}
};
【解题思路】
看似一道简单的题目,也不太简单。
关键点在于要按绝对值排序,再将所有小于0的数取反。剩余的k就只操作最小的数,而且是只判断是否是奇数,因为操作偶数次后的值不变。
134. 加油站
题目链接:134. 加油站,难度:中等
【实现代码】
class Solution {
public:
int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
int result = 0;
int rest = 0;
int sum = 0;
for (int i = 0; i < gas.size(); i++) {
rest += gas[i] - cost[i];
sum += gas[i] - cost[i];
if (rest < 0) {
result = i + 1;
rest = 0;
}
}
if (sum < 0) {
result = -1;
}
return result;
}
};
【解题思路】
贪心策略,从0索引开始,每次剩余流量累加的和不小于0时,则可以从0开始;
如果从在i处,累加的值小于0,则从第i+1个开始重新累加。
最后总的累加值小于0,则返回-1;否则可以从i+1处开始跑一圈。
135. 分发糖果
题目链接:135. 分发糖果,难度:困难
【实现代码】
class Solution {
public:
int candy(vector<int>& ratings) {
int result = 0;
vector<int> candies(ratings.size(), 0);
candies[0] = 1;
for (int i = 1; i < ratings.size(); i++) {
if (ratings[i] > ratings[i - 1]) {
candies[i] = candies[i - 1] + 1;
} else {
candies[i] = 1;
}
}
result = candies[ratings.size() - 1];
for (int i = ratings.size() - 2; i >= 0; i--) {
if (ratings[i] > ratings[i + 1]) {
candies[i] = max(candies[i], candies[i + 1] + 1);
}
result += candies[i];
}
return result;
}
};
【解题思路】
贪心思想,从前往后遍历一遍,再从后向前遍历一遍,但是从后向前更新糖果时,要比较当前有的糖果和后一个+1糖果的最大值。