现在的位置: 首页 > 综合 > 正文

[各种面试题] 分配糖果

2017年12月23日 ⁄ 综合 ⁄ 共 797字 ⁄ 字号 评论关闭

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;
}

抱歉!评论已关闭.