- 题目
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?
- 题意:给定一个序列,表示每个小朋友的等级大小,现在要给小朋友们分糖果,公平起见,对于相邻的小朋友,等级高的一定要比等级低的拿到更多的糖果。求最少一共要分多少的糖果。
思路:
- 因为只要关注相邻的小朋友,因此可以采用贪心策略,让等级高的总比等级低的多就行。那么就会出现一种呈山峰型的形状,先上后下,再上再下的形状。
- 那么现在只要确定好山峰的最佳大小,那么周围的就能够立即得到。而山峰的大小就取决于两边的长度,即max(left, right),而较短的那边只需要在最低点从1个糖果开始递增,即可得到最佳。
- 时间复杂度 O(N)
实现细节:
- 按照思路,从左到右,扫一次,默认先上后下,上升的过程中从1颗糖果开始往上递增,直到山峰,直接累计即可,而在下降过程中,要判断的是下降的长度,要注意max(left, right)这个条件即可。
- 要是直接按这个逻辑的话,实现起来比较复杂,边界条件容易错。因此,可以稍微调整一下,因为要保证的条件只有顶部位置为max(left, right),所以只需要从左上一次,从右上一次,而从右边的时候注意保证max(left, right)即可。用空间换取的实现简单正确。。
class Solution {
public:
int candy(vector<int>& ratings) {
int size=ratings.size();
vector<int> candies(size, 1);
for (int i=1;i<size;++i)
{
if (ratings[i] > ratings[i-1])
{
candies[i]=candies[i-1]+1;
}
}
for (int i=size-2;i>=0;--i)
{
if (ratings[i] > ratings[i+1])
{
candies[i] = max(candies[i], candies[i+1]+1);
}
}
int ans=0;
for (auto item: candies)
ans += item;
return ans;
}
};