leetcode-135.分发糖果

NLP菜鸟

于 2023-05-29 16:16:38 发布

阅读量129

点赞数

分类专栏： Leetcode错题集文章标签： leetcode 算法职场和发展

本文链接：https://blog.youkuaiyun.com/qq_45069496/article/details/130930226

版权

Leetcode错题集专栏收录该内容

21 篇文章

订阅专栏

该代码实现了一个基于贪心策略的算法，用于解决给定一组孩子评分，每个孩子至少要得到一个糖果，且相邻的孩子如果评分更高则应获得更多糖果的问题。算法首先初始化所有孩子糖果数为1，然后从左到右遍历，确保右侧孩子评分高时增加糖果数，接着从右到左遍历，修正左侧孩子可能少于右侧的情况。最后返回所有孩子的糖果总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
解题思路： 贪心（每次遍历只考虑一侧的大小关系）
（1）将所有孩子的糖果初始化为1
（2）从左往右遍历，若右边孩子的评分比左边的高，则右边孩子的糖果数更新为左边孩子的糖果数加1
（3）从右往左遍历，若左边孩子的评分比右边的高且左边孩子当前的糖果数不大于右边孩子的糖果数，则左边孩子的糖果数更新为右边孩子的糖果数加1

int candy(vector<int>& ratings) {
	int size = ratings.size();
	if (size < 2) {
		return size;
	}
	vector<int> num(size, 1);
	for (int i = 1; i < size; ++i) {
		if (ratings[i] > ratings[i-1]) {
			num[i] = num[i-1] + 1;
		}
	}
	for (int i = size - 1; i > 0; --i) {
		if (ratings[i] < ratings[i-1]) {
			num[i-1] = max(num[i-1], num[i] + 1);
		}
	}
	return accumulate(num.begin(), num.end(), 0); // std::accumulate 可以很方便地求和
}