算法学习---分发糖果

算法学习---分发糖果

老师想给孩子们分发糖果,有 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) 的解法,利用的是这里面的波峰波谷的思维,有兴趣的小伙伴可以在上面的官方链接中去查询,这里就不再阐述了。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值