题目描述
一群孩子站成一排,每一个孩子有自己的评分。现在需要给这些孩子发糖果,规则是如果一个孩子的评分比自己身旁的一个孩子要高,那么这个孩子就必须得到比身旁孩子更多的糖果;所有孩子至少要有一个糖果。求解最少需要多少个糖果。
输入输出样例
输入是一个数组,表示孩子的评分。输出是最少糖果的数量。
Input: [1,0,2]
Output: 5
在这个样例中,最少的糖果分法是 [2,1,2]。
题解1
两次遍历:
思路:两次遍历分别满足左右规则。
1.从左向右遍历时,保证序列中,右边分数较高的比相邻左边的糖果数多一。即:如果ratings[i]>ratings[i-1],则 candy[右=[i]=candy[i-1]+1 .此步骤保证了对于任意两个元素AB,若B>A,则B糖果数>A糖果数。
2.从右向左遍历时,保证序列中,左边分数较高的比相邻右边的糖果数多。如果ratings[i-1]>ratings[i],并且candy[i-1]<=candy[i],则candy[左]=candy[右]+1
提示:为什么从左往右遍历时,当右边比左边大时更新右边,而不是选择当左边比右边大时更新左边?
因为从左往右遍历,更新右边的值时,在下次循环时,能用到上次更新的值。
Java代码:
public int candy(int[] ratings) {
int numOfCandy = ratings.length;
int[] record = new int[ratings.length];
int cnt1 = 0;
int cnt2 = 0;
for(int i = 0; i < ratings.length-1; i++) {
if(ratings[i+1] > ratings[i]) {
numOfCandy += record[i]-record[i+1]+1;
record[i+1] = record[i]+1;
}
}
for(int i = ratings.length-1; i > 0; i--) {
if(ratings[i] < ratings[i-1] && record[i-1] <= record[i]) {
numOfCandy += record[i] - record[i-1] + 1;
record[i-1] = record[i] + 1;
}
}
return numOfCandy;
}
题解2
一次遍历:
思路:
当为升序时:后一个糖果数等于前一个+1
当为降序时:糖果增量为[降序序列长度];当降序序列长度大于增序序列长度时,增序序列最后一个也加入降序序列长度。
即:需要一个pre记录上一个人的糖果数,dec记录降序序列长度,ans记录总糖果数。
Java:
public int candy(int[] ratings) {
int numOfCandy = ratings.length;
int[] record = new int[ratings.length];
int cnt1 = 0;
int cnt2 = 0;
for(int i = 0; i < ratings.length-1; i++) {
if(ratings[i+1] > ratings[i]) {
numOfCandy += record[i]-record[i+1]+1;
record[i+1] = record[i]+1;
}
}
for(int i = ratings.length-1; i > 0; i--) {
if(ratings[i] < ratings[i-1] && record[i-1] <= record[i]) {
numOfCandy += record[i] - record[i-1] + 1;
record[i-1] = record[i] + 1;
}
}
return numOfCandy;
}