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

目录

题目链接：1005.K次取反后最大化的数组和

思路

代码

题目链接：134. 加油站

思路

代码

题目链接： 135. 分发糖果

思路

代码

总结

---

题目链接：1005.K次取反后最大化的数组和

思路

        数组可能出现以下几种情况：

        ①全为正。只对最小的数进行k次取反

        ②全为负。从最小（绝对值最大）的数开始取反，直至k个，或者到数组末尾，在对最后一个数字进行k-n次取反，n为数组元素个数

        ③正负都有（分别为a，b个，a+b=n）。将负的从绝对值最大到最小挨个取反，若k>b，对绝对值最小的进行k-b次取反；若k<b，与②类似。

        上述的思路正确，问题是如何实现。首先要对数组进行排序，但是排序规则是按照绝对值大小；其次是遍历数组，根据k对数组进行取反；最后再次从头遍历数组，对其元素加和。

代码

class Solution {
    // 定义排序规则，绝对值大的在前面
    static bool cmp(int a, int b) { return abs(a) > abs(b); }

public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end(), cmp);
        int sum = 0;
        for (int i = 0; i < nums.size() && k > 0; i++) {
            if (nums[i] < 0) {
                nums[i] = -nums[i];
                k--;
            }
        }
        // 数组遍历完毕，k还没用完，对绝对值最小的进行k次取反
        // 用k模取2，如果余数为1，说明需要对最小元素取反
        if (k % 2 == 1)
            nums[nums.size() - 1] *= -1;
        // 对修改后的数组进行元素加和
        for (int a : nums) {
            sum += a;
        }
        return sum;
    }
};

题目链接：134. 加油站

思路

        从当前站到下一站，要在当前站加油，然后到下一站的途中消耗油。把每站的加油量和消耗的油量做差，得到每站剩余的油量。

        如果无解，说明某一站剩余的油量为负数，且该负数的绝对值比其余站所有剩余油量的总和都要大，此时返回-1；

        如果有解，说明不会出现上述那样的负数，在某一站出发一定可以。只要从下标0开始统计，做累加，当遇到剩余油量之和为负数时，则令下一站为出发点。遍历一边数组进行更新，即可得到出发点。

代码

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, vector<int>& cost) {
        int total = 0;  // 所有站剩余油量之和
        int cursum = 0; // 出发站到当前站的剩余油量之和
        int start = 0;  // 起始位置
        for (int i = 0; i < gas.size(); i++) {
            int rest = gas[i] - cost[i]; // 当前站能够补给的油量
            total += rest;
            cursum += rest;
            // 出发站到当前站剩余油量为负，说明出发站选取不对
            // 更新出发站与出发站到当前站的剩余油量之和
            if (cursum < 0) {
                cursum = 0;
                start = i + 1;
            }
        }
        // 所有站的剩余油量之和小于0，无解
        if (total < 0)
            return -1;
        return start;
    }
};

题目链接：135. 分发糖果

思路

        根据分发规则，要比较每个孩子与左右两边的评分。分开比较：

        ①从左到右，判断右边孩子是否比左边孩子的评分高

        ②从右到左，判断左边孩子是否比右边孩子的评分高

        最后取二者的最大值，进行结果累加

代码

class Solution {
public:
    int candy(vector<int>& ratings) {
        vector<int> candys(ratings.size(), 1);
        // 从左到右，判断右是否比左高
        for (int i = 1; i < ratings.size(); i++) {
            if (ratings[i] > ratings[i - 1]) {
                candys[i] = candys[i - 1] + 1;
            }
        }
        // 从右到左，判断左是否比右高
        for (int i = ratings.size() - 2; i >= 0; i--) {
            if (ratings[i] > ratings[i + 1]) {
                // 跟第一次计算的糖果作比较，取最大值
                candys[i] = max(candys[i], candys[i + 1] + 1);
            }
        }
        int result = 0;
        for (int a : candys) {
            result += a;
        }
        return result;
    }
};

总结

        ①思路很重要，不要局限于题目的描述，然后对题目进行相应的模拟，应该跳出模拟，从本质下手，例如1055.K次取反后最大化的数组和，全为正数时，对最小元素不必重复取反，无非就是两种情况，取反和不变，只需要模取2即可；例如134.加油站，不用模拟从出发站走一圈的情况，首先将问题转变为剩余油量问题，其次也就两种情况，有解无解。无解时，说明剩余油量之和为负数；有解时，说明一定有一个站点可以当作出发站，所以只需要从头到尾遍历依次即可

        ②熟悉库函数。1055.K次取反后最大值的数组和中使用了sort函数，还限定了sort函数的排列规则。所以要熟悉常用的库函数的名称，函数类型，参数等

        ③比较问题，拆开来看。不要想着一步到位，而是拆分问题，多次处理。例如135.分发糖果，先从左到右比较，再从右到左比较，最后得到结果