代码随想录算法训练营第三十四天| 1005. K 次取反后最大化的数组和 & 134. 加油站 & 135. 分发糖果

代码随想录算法训练营第三十四天| 1005. K 次取反后最大化的数组和 & 134. 加油站 & 135. 分发糖果

---

1005. K 次取反后最大化的数组和

题目链接：1005. K 次取反后最大化的数组和 - 力扣（LeetCode）

题目描述：

​ 给你一个整数数组 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] 。

示例 3：

输入：nums = [2,-3,-1,5,-4], k = 2
输出：13
解释：选择下标 (1, 4) ，nums 变为 [2,3,-1,5,4] 。

提示：

• 1 <= nums.length <= 104
• -100 <= nums[i] <= 100
• 1 <= k <= 104

贪心策略：当存在负整数时，反转最小的负整数

​ 当不存在负整数时，反转最小的非负整数

class Solution {
private:
    static bool cmp(int a,int b){
        return abs(a)>abs(b);
    }
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        std::sort(nums.begin(),nums.end(),cmp);
        int ans = 0;

        for(int i = 0;i<nums.size();i++){
            if(nums[i]<0 && k>0){
                nums[i] *= -1;
                k--;
            } 
            ans+=nums[i];
        }
        if(k%2==1){
            nums[nums.size()-1] *= -1;
            ans += 2*nums.back(); 
        }
        return ans;
    }
};

---

134. 加油站

题目链接：134. 加油站 - 力扣（LeetCode）

题目描述：

​ 在一条环路上有 n 个加油站，其中第 i 个加油站有汽油 gas[i] 升。

​ 你有一辆油箱容量无限的的汽车，从第 i 个加油站开往第 i+1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发，开始时油箱为空。

​ 给定两个整数数组 gas 和 cost ，如果你可以按顺序绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1 。如果存在解，则 保证 它是 唯一 的。

示例 1:

输入: gas = [1,2,3,4,5], cost = [3,4,5,1,2]
输出: 3
解释:
从 3 号加油站(索引为 3 处)出发，可获得 4 升汽油。此时油箱有 = 0 + 4 = 4 升汽油
开往 4 号加油站，此时油箱有 4 - 1 + 5 = 8 升汽油
开往 0 号加油站，此时油箱有 8 - 2 + 1 = 7 升汽油
开往 1 号加油站，此时油箱有 7 - 3 + 2 = 6 升汽油
开往 2 号加油站，此时油箱有 6 - 4 + 3 = 5 升汽油
开往 3 号加油站，你需要消耗 5 升汽油，正好足够你返回到 3 号加油站。
因此，3 可为起始索引。

示例 2:

输入: gas = [2,3,4], cost = [3,4,3]
输出: -1
解释:
你不能从 0 号或 1 号加油站出发，因为没有足够的汽油可以让你行驶到下一个加油站。
我们从 2 号加油站出发，可以获得 4 升汽油。 此时油箱有 = 0 + 4 = 4 升汽油
开往 0 号加油站，此时油箱有 4 - 3 + 2 = 3 升汽油
开往 1 号加油站，此时油箱有 3 - 3 + 3 = 3 升汽油
你无法返回 2 号加油站，因为返程需要消耗 4 升汽油，但是你的油箱只有 3 升汽油。
因此，无论怎样，你都不可能绕环路行驶一周。

提示:

• gas.length == n
• cost.length == n
• 1 <= n <= 10^5
• 0 <= gas[i], cost[i] <= 10^4

解题策略：代码一是找到期间最大油量缺口，找到可以弥补缺口的那个加油站；代码二是当前油量始终保持为非负，一旦为负，则从下一个加油站开始计算

代码一：

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int sum = 0;
        int val = INT_MAX;
        for(int i = 0;i<gas.size();i++){
            sum += gas[i] - cost[i];
            if(sum<val) val = sum;
        }
        if(sum<0) return -1;
        if(val>=0) return 0;
        for(int i = gas.size()-1;i>=0;i--){
            val += gas[i] - cost[i];
            if(val >=0) return i;
        }
        return -1;
    }
};

代码二：当前优良始终保持为非负，一旦为负，则从下一个加油站开始计算

class Solution {
public:
    int canCompleteCircuit(std::vector<int>& gas, std::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;
    }
};

---

135. 分发糖果

题目链接：135. 分发糖果 - 力扣（LeetCode）

题目描述：

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

示例 1：

输入：ratings = [1,0,2]
输出：5
解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2：

输入：ratings = [1,2,2]
输出：4
解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
     第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。

提示：

• n == ratings.length
• 1 <= n <= 2 * 104
• 0 <= ratings[i] <= 2 * 104

class Solution {
public:
    int candy(vector<int>& ratings) {
        std::vector<int> can(ratings.size(),1);
        for(int i = 1;i<ratings.size();i++){
            if(ratings[i]>ratings[i-1]) can[i] = can[i-1]+1; 
        }
        for(int i = ratings.size()-2;i>=0;i--){
            if(ratings[i]>ratings[i+1]) {
                can[i] = std::max(can[i],can[i+1]+1);
            }
        }
        int ans = 0;
        for(auto x:can) ans += x;
        return ans;
    }
};

---