老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
leetCode题目链接:https://leetcode-cn.com/problems/candy
第一个想法
第一眼看到题目,似乎只要遍历一遍即可
举个例子:
可以从1-2-3这三步看出,如果是递减(下坡),每次要对递减序列都进行一次操作,
那么最差情况就是完全递减的序列,比如:5-4-3-2-1,需要进行n^2 次操作,
也就是最差时间复杂度为O(n^2) ,这显然不是最优情况。
第二个想法
从上面的图表看出,正序遍历并且为递增的部分,是不需要对整个递增序列进行多余操作的,
那么第二个想法也是呼之欲出了
我们只需要正序遍历一遍,取出递增的部分,再倒序遍历一遍,取出倒序时递增的部分即可,
看下面的动图理解,这个解法只需要一次正序遍历+一次倒序遍历,因此时间复杂度可以达到O(n)
贴一下第二想法的代码:
class Solution {
public:
int candy(vector<int>& ratings) {
if (ratings.size() == 1)
return 1;
vector<int> candies;
for (int i = 0; i < ratings.size(); ++i)
{
if (i == 0) candies.push_back(1);
else if (i == ratings.size() - 1) candies.push_back(ratings[i] > ratings[i - 1] ? candies[i - 1] + 1 : 1);
else candies.push_back(ratings[i] > ratings[i - 1] ? candies[i - 1] + 1 : 1);
}
for (int i = ratings.size() - 1; i >= 0; --i)
{
if (i == 0) candies[i] = (ratings[i] > ratings[i + 1] ? candies[i + 1] + 1 : 1);
else if (i == ratings.size() - 1) candies[i] = max(candies[i], 1);
else candies[i] = ratings[i] > ratings[i + 1] ? max(candies[i + 1] + 1, candies[i]) : candies[i];
}
int sum = 0;
for (int i = 0; i < candies.size(); ++i)
{
sum += candies[i];
}
return sum;
}
};
官方题解中还有一种空间复杂度为O(1) , 时间复杂度为O(n) 的解法,利用的是这里面的波峰波谷的思维,有兴趣的小伙伴可以在上面的官方链接中去查询,这里就不再阐述了。