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

贪心

---

一、K次取反后最大化的数组和

按照绝对值大小递减排序，优先反转为负数的元素，如果k还有剩余，则反转最后一个元素，也就是绝对值最小的元素

class Solution {
public:
    //按照绝对值排序，优先反转绝对值大的负数，如果k还有剩余则反转最后一个元素，也就是最小的元素
    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] = 0 - nums[i];
                k--;
            }
        }
        if (k % 2 == 1) nums[nums.size()-1] = 0 - nums[nums.size()-1];

        int sum = 0;
        for (auto i : nums) {
            sum += i;
        }
        return sum;
    }
};

二、加油站

1.如果cost-gas的总和大于等于0，说明可以环绕一圈，同时需要找出起点
2.curSum记录cost-gas，从0开始累加到i，如果curSum<0，说明[0,i]这一段不够燃油，不能环绕一圈，所以下次从i+1开始再次累加，start记录这个起点

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        //同时判断起点和是否能环绕一圈，如果总数大于0，说明能达到
        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) {
                curSum = 0;
                start = i + 1;
            }
        }
        if (totalSum < 0) return -1;
        return start;
    }
};

三、分发糖果

1.相邻两个孩子评分更高的获得更多的糖果，意思是要比相邻孩子的最多糖果数还要多一个，并不是在自己基础上加1
2.不仅要比较左边孩子，还要比较右边孩子。分为两个部分，首先从左到右比较左孩子，如何大于就比左孩子多一个糖果。接着从右到左，如果比右孩子大，则需要比较自身的糖果数量和右边孩子加一得到的糖果数量，取最大值，保证平衡

class Solution {
public:
    int candy(vector<int>& ratings) {
        //相邻两个孩子评分更高的获得更多的糖果，意思是要比相邻孩子的最多糖果数还要多一个，并不是在自己基础上加1
        vector<int>candyV(ratings.size(), 1);

        //从前往后，右孩子比较左孩子
        for (int i = 1; i < ratings.size(); i++) {
            if (ratings[i] > ratings[i-1]) candyV[i] = candyV[i-1] + 1;
        }

        //从后往前，左孩子比较右孩子
        //此时有两种选择，一种是candy[i+1]+1，另外一种是candyV[i]，取两者最大值，保证大于相邻孩子
        for (int i = ratings.size()-2; i >=0; i--) {
            if (ratings[i] > ratings[i+1]) candyV[i] = max(candyV[i], candyV[i+1] + 1);
        }

        int sum = 0;
        for (auto i : candyV) {
            sum += i;
        }
        return sum;
    }
};

---

总结

后面两题比较难想，还得多多练习
学习时间90min。
学习资料：《代码随想录》。