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

文章目录

  • 一、题目
  • 二、解法
  • 三、完整代码

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

一、题目

【算法与数据结构】135、LeetCode分发糖果_第1张图片

二、解法

  思路分析:本题的思路是要相比较一边,然后在比较另外一边,左右两边一起比较的代码非常难写。每个孩子的糖果数量初始化为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) O(n)
  • 空间复杂度: O ( n ) 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

你可能感兴趣的:(算法,算法)