leetcode Candy

最新推荐文章于 2023-02-28 14:31:32 发布

原创最新推荐文章于 2023-02-28 14:31:32 发布 · 386 阅读

0 ·

CC 4.0 BY-SA版权

leetcode Fighting 专栏收录该内容

133 篇文章

订阅专栏

本文介绍了一种高效的糖果分配算法，该算法通过三次遍历确保每个孩子的糖果数符合其评分等级，同时最小化总的糖果数量。首次从左到右遍历，确保当前孩子糖果数不低于左侧孩子并在评分更高时增加糖果；第二次从右向左遍历，确保右侧条件同样满足；最后遍历计算总糖果数。

此问题可以遍历三遍ratings解决：

第一遍遍历从左向右，若小孩 i 与其左边的小孩 i-1 满足如下: 若 i 的级别比 i-1 大，则candyNum[i] = candyNum[i-1]+1;

第二遍从右向左遍历，若小孩 i 与其右边的小孩 i+1 满足如下: 若 I 的级别比 i+1大且candyNum[i] <= candyNum[i+1] , 则candyNum[i] = candyNum[i+1]+1;

第三遍遍历求出总数

代码

class Solution
{
public:
	int candy(vector<int> &ratings)
	{
		vector<int> candyNum(ratings.size(), 1);

		for(int i = 1; i < ratings.size(); ++i)
		{
			if(ratings[i]>ratings[i-1])
			{
				candyNum[i] = candyNum[i-1]+1;
			}
		}


		int num = 0;

		for(int i = ratings.size()-2; i >= 0; --i)
		{
			if(ratings[i]>ratings[i+1]&&candyNum[i]<=candyNum[i+1])
				candyNum[i] = candyNum[i+1] + 1;
					
		}

		for(int i = 0; i < ratings.size(); ++i)
			num += candyNum[i];

		return num;
	
	}


};