【算法与数据结构】135、LeetCode分发糖果

所有的LeetCode题解索引，可以看这篇文章——【算法和数据结构】LeetCode题解。

一、题目

二、解法

  思路分析：本题的思路是要相比较一边，然后在比较另外一边，左右两边一起比较的代码非常难写。每个孩子的糖果数量初始化为1。程序当中我们首先从前往后遍历，若第i个孩子评分大于第i-1个孩子，其糖果数量为第i-1个孩子的糖果数量+1。然后从后往前遍历，若第i个孩子比第i-1个孩子和第i+1个孩子评分都高，则要拿到比傍边孩子更多的糖果，只需要在先前的candyVec[i]（先前的candyVec[i]数量必然大于第i-1个孩子的糖果数量）和第i+1个孩子糖果数量+1之间取最大值。
  程序如下：

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);	// 保证左边更大的孩子得到的糖果比左边和右边都多，取最大值
			}
		}
		return accumulate(candyVec.begin(), candyVec.end(), 0);
	}
};

复杂度分析：

• 时间复杂度： $O(n)$。
• 空间复杂度：$O(n)$。

三、完整代码

# include 
# include 
# include 
# include 
using namespace std;

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);	// 保证左边更大的孩子得到的糖果比左边和右边都多，取最大值
			}
		}
		return accumulate(candyVec.begin(), candyVec.end(), 0);
	}
};

int main() {
	vector<int> ratings = { 1,2,2 };		// 加油站的油量
	Solution s1;
	int result = s1.candy(ratings);
	cout << result << endl;
	system("pause");
	return 0;
}

end