标签 |
---|
贪心 |
题目
https://leetcode-cn.com/problems/candy/
1.
分析
先从左往右遍历,再从右往左遍历,取其中较大值。
代码
class Solution {
public int candy(int[] ratings) {
if(ratings==null||ratings.length==0)
return 0;
int len=ratings.length;
int[] left=new int[len];
int[] right=new int[len];
Arrays.fill(left,1);
Arrays.fill(right,1);
for(int i=1;i<len;i++){
if(ratings[i-1]<ratings[i])
left[i]=left[i-1]+1;
}
int count=left[len-1];//最后一个的数目
for(int i=len-2;i>=0;i--){
if(ratings[i+1]<ratings[i])
right[i]=right[i+1]+1;
count+=Math.max(left[i],right[i]);
}
return count;
}
}
复杂度
时间复杂度 O(N): 遍历两遍数组即可得到结果;
空间复杂度 O(N) : 需要借用left,right的线性额外空间。
结果
2.常数空间
分析
pre:前一个同学分得的糖果数量
dec:当前递减序列的长度
inc:最近的递增序列的长度
代码
class Solution {
public int candy(int[] ratings) {
if(ratings==null||ratings.length==0)
return 0;
int ret=1;
int dec=0;//当前递减序列的长度
int inc=1;//最近的递增序列的长度
int pre=1;//前一个同学分得的糖果数量
for(int i=1;i<ratings.length;i++){
if(ratings[i]>=ratings[i-1]){
dec=0;//出现了递增,所以当前递减序列长度为0
pre=ratings[i]==ratings[i-1]?1:pre+1;//如果大于ratings[i-1],糖果数是前一个pre加一
inc=pre;//最近的递增序列长度=当前位置糖果数,跟上一行代码有关。如果i=i-1,那么当前递增长度重新回到1
ret+=pre;//总糖果数+=最近的递增序列长度
}
else{
dec++;//当前递减序列长度+1
//有一个既在递增序列也在递减序列,,也需要再分一个糖果,加进递减序列
if(dec==inc){
dec++;
}
ret+=dec;//dec=当前递减序列长度=递减序列中所有的同学都再多分配一个糖果
pre=1;
}
}
return ret;
}
}
复杂度
时间复杂度:O(n),其中n 是孩子的数量。我们需要遍历两次数组以分别计算满足左规则或右规则的最少糖果数量。
空间复杂度:O(1)。我们只需要常数的空间保存若干变量。