Day34【贪心算法】1005.K次取反后最大化的数组和、134.加油站、135.分发糖果

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

力扣题目链接/文章讲解

视频讲解

本题需要用到两次贪心：

第一次贪心：当数组中还有负数的时候，优先取反绝对值最大的负数 

第二次贪心：当数组中无负数时，优先取反最小的非负数 

第一次贪心如何实现：对数组按照绝对值由大到小排序，然后依次遍历处理

第二次贪心如何实现：数组已经按照绝对值由大到小排序了，且数组中无负数，则最小的非负数就是数组的末尾元素

代码如下

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        // 先按照绝对值大小从大到小排序，用了lambda表达式所生成的可调用对象
        sort(nums.begin(), nums.end(), [](int lhs, int rhs) -> bool { return abs(lhs) > abs(rhs); });
        for (auto & i : nums) { // 注意这里要用引用
            if (k > 0 && i < 0) {   // 如果还有取反次数，则取反绝对值最大的负数
                i = -i;
                --k;    // 消耗一次取反次数
            }
        }
        if (k % 2 == 1) nums.back() *= -1;  // 如果还剩取反次数且为奇数，则不断取反最小的非负数，最后相当于取反了一次
        // 如果没剩取反次数了或者取反次数为偶数，则偶数次对一个元素取反等于0次取反
        return accumulate(nums.begin(), nums.end(), 0);
    }
};

134.加油站

力扣题目链接/文章讲解

视频讲解

首先计算每个加油站的剩余量（将到达每个加油站的加油和离开该加油站的消耗相减，得到rest[i]为gas[i] - cost[i]）

只要每个加油站剩余量的和大于等于0，就存在能够环路行驶一周的出发方案（这是直观常识）

现在关键是找出发点

贪心过程：i从0开始累加rest[i]，和记为curSum，一旦curSum小于零，说明[0, i]区间都不能作为起始位置，因为这个区间选择任何一个位置作为起点，到i这里都会断油，那么起始位置从i+1算起，再从0计算curSum 

代码如下：

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        vector<int> rest;
        for (int i = 0; i < gas.size(); ++i) {  
            rest.push_back(gas[i] - cost[i]);   // 计算每个加油站的剩余量
        }   
        if (accumulate(rest.begin(), rest.end(), 0) < 0) return -1; // 剩余量的和小于0，一定没办法走完一圈
        // 剩余量的和大于0，一定能走完一圈，剩下就是找起始点
        int start = 0;  // 记录起点
        int curSum = 0;
        for (int i = 0; i < rest.size(); ++i) {
            curSum += rest[i];
            if (curSum < 0) {   // 断油，起始位置从i+1算起，再从0计算curSum
                start = i + 1;
                curSum = 0; 
            }
        }
        return start;
    }
};

这道题没做过想不到 

135.分发糖果 

力扣题目链接/文章讲解

视频讲解

每个孩子首先初始化为至少一个糖果

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);
    }
}

注意：如果 ratings[i] > ratings[i + 1]，此时candyVec[i]（第i个小孩的糖果数量）就有两个选择了，一个是candyVec[i + 1] + 1（从右边这个加1得到的糖果数量），一个是candyVec[i]（之前比较右孩子大于左孩子得到的糖果数量） 

candyVec[i]只有取最大的才能既保持对左边candyVec[i - 1]的糖果多，也比右边candyVec[i + 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;
    }
};

---

回顾总结 

贪心思路的选取很关键，也很难想