老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
int candy(int* ratings, int ratingsSize){
if(ratingsSize==0)return 0;
int *count=(int *)calloc(ratingsSize,sizeof(int));
count[0]=1;
for(int start_pos=0,min_pos,end_pos;start_pos<ratingsSize-1;start_pos=end_pos)
{
min_pos=start_pos-1;
for(int i=start_pos;i<ratingsSize;i++)
{
if(i>start_pos&&ratings[i]>=ratings[i-1])break;
min_pos++;
}
end_pos=min_pos-1;
bool the_same=false;
for(int i=min_pos;i<ratingsSize;i++)
{
if(i>min_pos)
{
if(ratings[i]<ratings[i-1])break;
else if(ratings[i]==ratings[i-1])
{
the_same=true;
break;
}
else {}
}
end_pos++;
}
count[min_pos]=1;
for(int i=min_pos-1;i>start_pos;i--)
{
count[i]=count[i+1]+1;
}
if(min_pos!=start_pos)count[start_pos]=count[start_pos]>count[start_pos+1]+1?count[start_pos]:count[start_pos+1]+1;
for(int i=min_pos+1;i<=end_pos;i++)
{
count[i]=count[i-1]+1;
}
if(the_same)
{
end_pos++;
if(end_pos<ratingsSize)count[end_pos]=1;
}
}
int ret=0;
for(int i=0;i<ratingsSize;i++)ret+=count[i];
free(count);
return ret;
}
执行用时 :32 ms, 在所有C提交中击败了30.30% 的用户
内存消耗 :8.7 MB, 在所有C提交中击败了8.70%的用户