[各种面试题] 分配糖果

n个小朋友站成一排(编号从0n-1),每个小朋友有一个rating值,存放在ratings数组中。老师需要给他们分配糖果,每个小朋友至少需要一颗糖果,对于任意相邻的两个小朋友ii+1,rating值大的必须比rating值小的分配的糖果多(rating相同的没必要分配一样多的糖果)。

请计算最少需要多少颗糖果,才能完成上述分配。

提示:本题存在O(n)的算法。

题目来源:CareerCup和InterviewStreet。

比较trick的题,其实想想就是连续递增或者递减序列的长度。

对第i个小朋友来说,如果往左连续有left个小朋友ratings比他小,所以这left个小朋友每个都要比左边个至少多拿一个糖果,所以第i个小朋友至少要拿left个糖果,对吧;

往右看也是一样的,所以对第i个小朋友至少要max(left[i],right[i])个糖果,最后统计就好了。

//返回最少需要的糖果数
long long minimalCandies(vector<int> &ratings) {
	if ( ratings.empty() ) 
		return 0;
	vector<long long> left(ratings.size(),1);
	vector<long long> right(ratings.size(),1);
	for(int i=1;i<ratings.size();i++)
	{
		if ( ratings[i]>ratings[i-1])
			left[i]=left[i-1]+1;
	}
	for(int i=ratings.size()-2;i>=0;i--)
	{
		if ( ratings[i]> ratings[i+1])
			right[i]=right[i+1]+1;
	}
	long long ans=0;
	for(int i=0;i<ratings.size();i++)
		ans+=max(left[i],right[i]);
	return ans;
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值