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

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

贪心局部最优：将绝对值大的负数变为正数，当前和变为最大；全局最优：整体获得最大和。
如果负数都变成正数之后，k > 0，仍然需要继续翻转，贪心局部最优：将最小的正数翻转，当前和变为最大；全局最优：整体获得最大和。
按绝对值排序很重要！

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);
		for(int i = 0; i < nums.size(); i++){
			if(nums[i] < 0 && k > 0){
				nums[i] *= -1;  // 将负数翻转
				k--;
			}
		}
		if(k % 2)  nums[nums.size() - 1] *= -1;  // 如果仍然需要继续翻转，将最小正数翻转
		int result = 0;
		for(int i : nums)  result += i;
		return result;
	}
};

加油站

最好想的就是暴力求解，遍历每一个加油站作为起点，看能否找到合适的解。
方法一：
先从下标0加油站出发。
情况1：gas总量比cost总量小，净消耗，肯定跑不到一圈
情况2：整个遍历过程中没出现过余量累加和小于0的情况，那么0就是起点
情况3：从后向前遍历，累加剩余油量，累加和能弥补从0出发的产生的最大亏空，那么从该节点就一定能回来，因为一定不会出现累加和小于0的情况了。

class Solution{
public:
	int canCompleteCircuit(vector<int>& gas, vector<int>& cost){
		int curSum = 0;
		int min = INT_MAX;
		for(int i = 0; i < gas.size(); i++){
			curSum += gas[i] - cost[i];
			if(min > curSum){
				min = curSum;
			}
		}
		if(curSum < 0)  return -1;
		if(min >= 0)  return 0;
		for(int i = gas.size() - 1; i > 0; i--){
			min += gas[i] - cost[i];
			if(min >= 0){
				return i;
			}
		}
		return -1;
	}
};

方法二：最好理解的贪心法
如果总耗油数目大于加油数目，那么不可能走完一圈。如下图所示，从下标0开始累加剩余量数组，和记为 curSum。一旦 curSum < 0，说明[0, i]区间内都没法作为起点走完一圈。继续从 i + 1 开始累计剩余油量数组，重复上述过程。

局部最优：余量 rest[i] 的累加和 curSum<0 时，起始位置至少在 i + 1。因为 i 之前的位置都一定不能跑完一圈。
全局最优：找到能跑一圈的起点。

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];  // 也可以在前面独立写一个for循环计算总余量然后判断，这里将两个for循环合为一个，写法上更简洁
			if(curSum < 0){
				curSum = 0;
				start = i + 1;
			}
		}
		if(totalSum < 0)  return -1;
		return start;
	}
};

分发糖果

双向的处理思想是关键，要同时考虑一个小孩的两侧并不容易。先保证右侧评分高的孩子会比左侧评分低的孩子得到更多的糖果，再去考虑另一侧，这样思路会清晰很多。
第一次遍历的局部最优：只要右边评分比左边大，右边的孩子就比左边多一个糖果；全局最优：评分高的右孩子比左侧孩子获得更多的糖果。（从前向后遍历）
第二次遍历要考虑左侧孩子评分比右侧高的情况。局部最优：只要左边评分比右边大，左边的孩子就比右侧多一个糖果；全局最优：评分高的左孩子比右侧孩子获得更多的糖果。（从后向前遍历）
取两次遍历的最大值，这样可以保证评分更高的孩子可以得到比相邻左孩子和右孩子都多的糖果。

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 < ratings.size(); i++){
			result += candyVec[i];
		}
		return result;
	}
};