1005. K 次取反后最大化的数组和
题目
给你一个整数数组 nums 和一个整数 k ,按以下方法修改该数组:
选择某个下标 i 并将 nums[i] 替换为 -nums[i] 。
重复这个过程恰好 k 次。可以多次选择同一个下标 i 。
以这种方式修改数组后,返回数组 可能的最大和 。
示例 1:
输入:nums = [4,2,3], k = 1
输出:5
解释:选择下标 1 ,nums 变为 [4,-2,3] 。
示例 2:
输入:nums = [3,-1,0,2], k = 3
输出:6
解释:选择下标 (1, 2, 2) ,nums 变为 [3,1,0,2] 。
解题思路
- 返回的数据是数组的和,不是数组元素,因此可以对输入的数组进行排序。
- 按照贪心的思路,排序后,一直换小值即可
- 这里需要注意的一点是,数组有小于0的数,如果想让和最大,还需要尽可能的把负数都更换为正数
- 这里给出的解题思路是:
- 将数组从达到小排序,找到
数组中最大的小于0的数
的下标。 - 然后,依次把小于0的数组元素切换为正数。并K–
- 如果替换到最后,K还是大于0,也即还有切换的次数,那么若K为奇数则替换一次,去反即可。若K为偶数,那替换K次还是本身,就不用再替换了。
那么本题的解题步骤为:
第一步:将数组按照绝对值大小从大到小排序,注意要按照绝对值的大小
第二步:从前向后遍历,遇到负数将其变为正数,同时K–
第三步:如果K还大于0,那么反复转变数值最小的元素,将K用完
第四步:求和
代码
class Solution {
static bool cmp(int a, int b) {
return abs(a) > abs(b);
}
public:
int largestSumAfterKNegations(vector<int>& A, int K) {
sort(A.begin(), A.end(), cmp); // 第一步
for (int i = 0; i < A.size(); i++) { // 第二步
if (A[i] < 0 && K > 0) {
A[i] *= -1;
K--;
}
}
if (K % 2 == 1) A[A.size() - 1] *= -1; // 第三步
int result = 0;
for (int a : A) result += a; // 第四步
return result;
}
};
135. 分发糖果
题目
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
示例 1:
输入:ratings = [1,0,2]
输出:5
解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
示例 2:
输入:ratings = [1,2,2]
输出:4
解释:你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这满足题面中的两个条件。
解题思路
- 按照贪心的思路
- 我至少有一个糖
- 我左边的孩子评分比我高,那左边的孩子比我多一个糖果 - 从后向前
- 我右边的孩子评分比我高,那右边的孩子比我多一个糖果 - 从前向后
- 先从前向后比较,即每次和我右边的孩子比较,如果右边的孩子评分比我高,那多一个糖果
- 然后从后向前比较,如果左边的孩子评分比我高,那多一个糖果。(这里的“我”指的是每次循环的当前节点)
- 左孩子大于右孩子的情况一定要从后向前遍历
代码
class Solution {
public:
int candy(vector<int>& ratings) {
vector<int> candyVec(ratings.size(), 1);
// 从前向后
for (int i = 1; i < ratings.size(); i++) {
if (ratings[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);
}
}
// 统计结果
int result = 0;
for (int i = 0; i < candyVec.size(); i++) result += candyVec[i];
return result;
}
};