There are N children standing in a line. Each child is assigned a rating value.
You are giving candies to these children subjected to the following requirements:
- Each child must have at least one candy.
- Children with a higher rating get more candies than their neighbors.
What is the minimum candies you must give?
常规解决方法:使用2个数组left和right来存储当前位置的左侧和右侧比其小的元素个数,最后分发糖果时,取最大的值便是当前所应该分发的糖果数,由于糖果数最少为1,而实现时最小的为0,故在最后应该加上所有的孩子数。
时间复杂度:O(N)
空间复杂度:O(N)
应该考虑到相邻孩子的分数一样时,怎么处理?题目是将其视为和评分小于当前位置一样处理。
<span style="font-size:14px;"> public int candy(int[] ratings) {
//从左边开始扫描记录每个位置左边比它小的元素的个数
int len=ratings.length;
int[] left=new int[len];
int[] right=new int[len];
for(int i=1;i<len;i++){
if(ratings[i]>ratings[i-1])
left[i]=1+left[i-1];
else//包含相等的评分处理
left[i]=0;
}
for(int i=len-2;i>=0;i--){
if(ratings[i]>ratings[i+1])
right[i]=1+right[i+1];
else
right[i]=0;
}
int result=0;
for(int i=0;i<len;i++){
result+=left[i]>right[i]?left[i]:right[i];
}
return result+len;
}</span>
优化解决办法:
时间复杂度:O(N)
空间复杂度:O(1)
思想:从前往后扫描,用2个变量prev,count,prev用来记录评分递增序列最后一个序列分配的糖果数(只考虑到从开始到现在),count用来记录递减序列的个数,当当前评分大于前一项的时候,若是count大于0,表明该项之前是递减序列,此时就可以确定递减序列应该分配的糖果数,即count*(count+1)/2,1,2,。。。,count;此时还应该考虑递减序列之前的那个位置的糖果数,即比较count和prev的大小,因为当时分配糖果的时候,只考虑了前面比其小的个数,现在应该更新;还有一点要注意当当前评分和前一项评分一样时应该如何处理。
最后还应该考虑最后几项都是递减时,应该加上相应的糖果数。
public int candy(int[] ratings) {
//从左边开始扫描记录每个位置左边比它小的元素的个数
int len=ratings.length;
int prev=1;
int count=0;//存储分数连续降低的个数
int result=1;
for(int i=1;i<len;i++){
if(ratings[i]>=ratings[i-1]){
if(count>0){//表明之前的都是递减的,计算递减的序列应该分的糖果数
result+=count*(count+1)/2;
if(count>=prev)
result+=count-prev+1;
count=0;
prev=1;
}
//计算当前位置应该分的糖果数
if(ratings[i]==ratings[i-1]){
result+=1;
prev=1;
}else{
result+=prev+1;
prev++;
}
}else{
count++;
}
}
if(count>0){//最后的几个是递减的
result+=count*(count+1)/2;
if(count>=prev)
result+=count-prev+1;
}
return result;
}