代码随想录day33 贪心算法进阶

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

题目

给定一个整数数组 A，我们只能用以下方法修改该数组：我们选择某个索引 i 并将 A[i] 替换为 -A[i]，然后总共重复这个过程 K 次。（我们可以多次选择同一个索引 i。）

以这种方式修改数组后，返回数组可能的最大和。

示例 1：

• 输入：A = [4,2,3], K = 1
• 输出：5
• 解释：选择索引 (1,) ，然后 A 变为 [4,-2,3]。

思考

这题其实非常巧妙，我们得用一个特殊的排序方法：把数组按照绝对值从大到小排序，然后遍历数组，碰到负数就把它变成正数，最后如果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] *= -1;

                k--;

            }

        }

        if(k % 2 == 1) nums[nums.size() - 1] *= -1;

        int res = 0;

        for(auto n : nums) res += n;

        return res;

    }

};

134. 加油站

题目

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

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

如果你可以绕环路行驶一周，则返回出发时加油站的编号，否则返回 -1。

说明:

• 如果题目有解，该答案即为唯一答案。
• 输入数组均为非空数组，且长度相同。
• 输入数组中的元素均为非负数。

示例 1: 输入:

• gas = [1,2,3,4,5]
• cost = [3,4,5,1,2]

思考

有点难这题，像动态规划，有几个关键需要想明白：

1、如果gas的总和小于cost的话，那么从哪一点出发都不可能，所以我们先假定gas的总和大于cost，但每一次要把当前gas和cos和的差值记录下来，记为totoalSum

2、我们就假定从第一个gas出发，把邮箱当前剩余的油记为curSum，如果说curSum<0，那么前面所有的记录都放弃，curSum = 0，从下一个加油站重新开始，这里体现了贪心算法的一个思考习惯就是，以局部代替整体，即局部不符合条件那么将局部带入到整体肯定不符合条件，极端情况就是如果curSum值记录一个加油站，那么当它小于0时肯定要把这个加油站放弃

3、用一个start来记录最终从哪个加油站开始，即如果curSum小于0，那么start = i+1

代码

class Solution {

public:

    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {

        int curSum = 0;

        int totalSum = 0;

        int index = 0;

        for(int i = 0; i < cost.size(); i++) {

            curSum += (gas[i] - cost[i]);

            totalSum += (gas[i] - cost[i]);

            if(curSum < 0) {

                index = i + 1;

                curSum = 0;

            }

        }

        if(totalSum >= 0) return index;

        else return -1;

    }

};

135. 分发糖果

题目

老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。

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

• 每个孩子至少分配到 1 个糖果。
• 相邻的孩子中，评分高的孩子必须获得更多的糖果。

那么这样下来，老师至少需要准备多少颗糖果呢？

示例 1:

• 输入: [1,0,2]
• 输出: 5
• 解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。

思考

这题很好玩，因为如果一个孩子如果比他左边大也比右边大，那么他得到的糖果需要是左右两边孩子的得到多的糖果的那一方再加1，说起来有点绕，举例就是：[1,3,2,1]，这个3的孩子得到的糖果是3，因为他左边孩子得到的是1，右边孩子得到的是2，那么按此逻辑，我们就需要从前往后遍历一遍再从后往前再遍历一遍，这样才能得到准确值

代码

class Solution {

public:

    int candy(vector<int>& ratings) {

        int res = 0;

        vector<int> candy(ratings.size(), 1);//首先将所有candy置为1

        // 从左向右遍历一遍再从右向左遍历一遍

        for(int i = 1; i < ratings.size(); i++) {

            if(ratings[i] > ratings[i-1]) candy[i] = candy[i-1] + 1;

        }

        for(int i = ratings.size() - 2; i>=0; i--) {

            if(ratings[i] > ratings[i+1]) {

                candy[i] = max(candy[i+1] + 1, candy[i]);//注意，这里要用max，因为可能存在candy[i]很大的情况，比如rating是[1,2,3,4,5,4]

            }

        }

        for(auto i : candy) res += i;

        return res;

    }

};