题目描述
有n(n<=100000)个小朋友站成一排(编号从0到n-1),每个小朋友有一个rating值,存放在ratings数组中。老师需要给他们分配糖果,每个小朋友至少需要一颗糖果,对于任意相邻的两个小朋友i和i+1,rating值大的必须比rating值小的分配的糖果多(rating相同的没必要分配一样多的糖果)。
请计算最少需要多少颗糖果,才能完成上述分配。
题目在线练习地址:http://www.itint5.com/oj/#31
这道题最开始的思路是将ratings排序,将其下标顺序存储在sort数组中,但是由于先排序,再更新,时间复杂度太高。
于是放弃排序,而是正序遍历一遍,查看相邻数是否反序,再反序遍历一遍,查看相邻数是否反序。这样在O(n)的时间复杂度下就完成了分配。
public class Candy {
// 返回最少需要的糖果数
public long minimalCandies(int[] ratings) {
int n = ratings.length;
if(n == 0){
return 0;
}
long sum = 0;
int[] count = new int[n];
for(int i = 0; i < n; i++){
count[i] = 1;
}
for(int i = 0; i < n-1; i++){
if(ratings[i]>ratings[i+1]){
if(count[i]<=count[i+1]){
count[i] = count[i+1]+1;
}
}else if(ratings[i]<ratings[i+1]){
if(count[i]>=count[i+1]){
count[i+1] = count[i]+1;
}
}
}
for(int i = n-1; i > 0; i--){
if(ratings[i]>ratings[i-1]){
if(count[i]<=count[i-1]){
count[i]= count[i-1]+1;
}
}else if(ratings[i]<ratings[i-1]){
if(count[i]>=count[i-1]){
count[i-1]=count[i]+1;
}
}
}
for(int i = 0; i < n; i++){
sum = sum+ count[i];
}
return sum;
}
}