题目:
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?
题意:
有N个小孩围坐一圈,每个小孩有个数值。现在需要给小孩糖。
每个小孩最少一个糖。
小孩子拥有更高的值就应该拥有更多的糖。
找出最少需要几颗糖。
思路:
引用”水中的鱼“博客中的图,如下:
我们可以把每一个递增,每一个递减区分开来。但是会出现的情况有两种:由于极大值点比较小(比如上图中左边例子的极大值是2),导致到达谷底时,出现了负值,或者到达谷底时出现的值比1大。
先将每一个孩子设定为只给一颗糖(即每个孩子都有可能是在谷底),那么我们从左往右扫描,只考虑递增的情况,遇到递增的就将该孩子的糖设置为前者糖的数量加1,也就是说这样子从左往右扫描一遍可以保证谷底的元素不会是负数,谷底的元素一定是1。上图左边的例子就会变成1,2,1,1,1,但是很明显,第三第四个元素应该比最后一个元素来的大。这时候开始修正,从最右边往最左边扫描,如果是递增的,那么按照递增的顺序改变值,那么上面的例子就会变成1,4,3,2,1,第二,三,四个元素都需要修正。我们再考虑右边的那个例子。第一遍才从左到右扫描完糖的数目是1,2,3,4,1,1,2,然后从右往左修正,得到结果是把第五个数的1,变成比第六个数大1,即2,所以结果是1,2,3,4,2,1,2
代码如下:
class Solution {
public:
int candy(vector<int>& ratings) {
int size = ratings.size();
if(size <= 1)return ratings.size();
vector<int> nums(size, 1);
for(int i = 1; i < size; i++) {
if(ratings[i] > ratings[i-1])
nums[i] = nums[i-1] + 1;
}
for(int i = size - 1; i > 0; i--) {
if(ratings[i-1] > ratings[i])
nums[i-1] = max(nums[i-1], nums[i] + 1);
}
int result = 0;
for(int i = 0; i < size; i++)
result += nums[i];
return result;
}
};