Candy

最新推荐文章于 2024-09-18 07:45:00 发布

原创最新推荐文章于 2024-09-18 07:45:00 发布 · 423 阅读

0 ·

CC 4.0 BY-SA版权

Leedcode 专栏收录该内容

72 篇文章

订阅专栏

博客围绕给N个孩子分糖果的问题展开，要求每个孩子至少有一个糖果，且评分高的孩子比邻居糖果多。给出解题思路，先从左到右扫描，再从右到左扫描，取每个孩子糖果数的最大值，同时保证每个小孩至少有一个糖果。

There are N children standing in a line. Each child is assigned a rating value.

You are giving candies to these children subjected to the following requirements:
?Each child must have at least one candy.
?Children with a higher rating get more candies than their neighbors.
What is the minimum candies you must give?

思路：

1.从左到右扫描，波谷为1，然后依次加1到波峰，再到波谷又是1.这样就能够保证，rating高的比左边rating低的糖果都要多。

2.从右到左扫描，同上。因为之前已经算过了依次，对于每个孩子要取candy[i]的最大，才能同时满足两个条件。

3,另外还要保证每个小孩都至少有一个糖果。

class Solution{
public:
	static vector<int> candy(vector<int>&ratings)
	{
		const int n = ratings.size();
		vector<int> candy(n);
		
		//从左向右扫描，遇到波谷设置为1，往波峰逐渐加1，到波谷有设置为1.这样保证，每rating高的要比rating低的糖果多。
		for(int i=1,incr=1;i<n;i++)
		{
			if (ratings[i] > ratings[i-1])
				candy[i] = max(incr++,candy[i]);
			else
				incr = 1;
		}

		//从右向左扫描。因为之前已经算过了依次，对于每个孩子要取candy[i]的最大，才能同时满足两个条件。
		for(int i = n-2,incr=1;i>=0;i--)
		{
			if(ratings[i] > ratings[i+1])
				candy[i] = max(incr++,candy[i]);
			else
				incr = 1;
		}

		//上面candy结果计算了步长。对于谷底无论从左还是从右，candy都不会赋值，会是0。因为每个小孩至少一个，所以需要加1。
		for(int i=0;i<candy.size();i++)
			candy[i]+=1;

		return candy;

		//计算糖果的总和
		//for(int i =0;i < n;i++)
		//{
			//sum += candy[i];
		//}

		//return sum;
	}
};