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

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

题目链接：1005

方法1

思路：每次都取反最小的数。当存在负数时，最小的数绝对值最大，反转可以得到最大的正数；当都为正时，反转最小的数可得绝对值最小的负数。
代码：

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        int result = 0;
        for (int i=0; i<k; i++) {
            vector<int>::iterator min_it = min_element(nums.begin(), nums.end());
            *min_it = -*min_it;
        }
        for (int i=0; i<nums.size(); ++i) {
            result += nums[i];
        }
        return result;
    }
};

方法2

思路：将所有数按绝对值由大到小排序。先从头将负数取反，这样保证越小的负数越先取反。若所有负数取反后，次数小于k，则接下来就是将最小的正数不断取反直至次数等于k。
代码：

class Solution {
public:
    int largestSumAfterKNegations(vector<int>& nums, int k) {
        sort (nums.begin(), nums.end(), [](int x, int y){
            return abs(x) > abs(y);
        });
        for (int i=0; i<nums.size(); ++i) {
            if (nums[i] < 0) {
                nums[i] *= -1;
                k--;
            }
            if (k <= 0) break;
        }
        if (k%2 == 1) nums[nums.size()-1] *= -1;
        int result = 0;
        for (auto a: nums) result += a;
        return result;
    }
};

LeetCode 134 加油站

题目链接：134
思路：gas[i]-cost[i]可得从i开到i+1的汽油净消耗rest[i] 。若rest[i]的总和大于等于0，则说明存在解。此时，从0开始，累加rest[i]为curSum。（1）假设加到k时，curSum<0，说明从[0,k]中的任一点出发，到达k时一定会使燃油耗尽，起始位置只能从k+1开始，curSum重置为0。（2）当从某一位置l开始，到末尾的累加一直大于0，则说明l一定是可行解。
证明：（1）假设存在j在[0,k)之间使得[j,k]的累加大于0，由于[0,k]的累加小于0，因此[0,j]的累加小于0，这与到k时curSum才小于0矛盾。（2）从l开始，则[0,l-1]不论分了几段，累加一定小于0。当存在解时，rest[i]的总和大于等于0，因此[l,n-1]的累加的绝对值一定大于[0,l-1]的累加绝对值，则从l开始一定能满足要求。
代码：

class Solution {
public:
    int canCompleteCircuit(vector<int>& gas, 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) {
                curSum = 0;
                start = i+1;
            }
        }
        return totalSum<0 ? -1 : start;
    }
};

LeetCode 135 分发糖果

题目链接：135
思路：左右情况分开考虑。先从头遍历，考虑右边评分大于左边的情况：若右边评分比左边大，则右边的糖果比左边多1，否则给1个糖果。然后从尾部开始遍历，考虑左边评分大于右边的情况：若左边评分大于右边，即rating[i]>rating[i+1]，则取max(candy[i],canding[i+1]+1)，保证当rating[i]大于左右两边时，candy[i]也大于左右两边。
代码：

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