您的位置:首页 > 职场人生

[各种面试题] 分配糖果

2013-09-19 21:20 197 查看
有n个小朋友站成一排(编号从0到n-1),每个小朋友有一个rating值,存放在ratings数组中。老师需要给他们分配糖果,每个小朋友至少需要一颗糖果,对于任意相邻的两个小朋友i和i+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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: