1、题目描述
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
2、算法分析
ratings
数组表示孩子们的评分。要求:
①每个孩子至少分发一个糖果
②相邻孩子之间评分高的会获得更多的糖果。
返回的结果是最少分发的糖果数目。
思路:
红色代表的是从前往后,绿色代表的是从后往前。
总体思路:
每个孩子至少分发一个糖果,创建一个糖果的数组,也就是结果集的数组,初始化数组,初始化数组的元素都为1,然后从左到右遍历数组;
从前往后遍历,如果当前元素大于其相邻左边的元素,那么result[i] = result[i - 1] + 1;
从后往前遍历,如果当前元素小于其相邻右边的元素,那么result[i] = result[i + 1] + 1;
详细思路:
①每个孩子至少分发一个糖果,初始化数组元素为1
②从左到右遍历,因为糖果的数组初始化为1了,只需要判断评分是否比其左边相邻的元素大,大的话result[i] = result[i - 1] + 1
③从右到左遍历,因为第一遍,从左到右进行遍历后,糖果数组的元素值已经改变了。如果当前元素比相邻右边的元素大的话, 且糖果值比result[j] <= result[j + 1]小,result[j] = result[j + 1] + 1;
④计算总糖果数的和。
3、代码实现
class Solution {
public int candy(int[] ratings) {
int[] result = new int[ratings.length];
int count = 0;
for(int i = 0;i < result.length;i++){
result[i] = 1;
}
// 从前往后
for(int i = 1;i < result.length;i++){
if(ratings[i] > ratings[i - 1]){
result[i] = result[i - 1] + 1;
}
}
// 从后往前遍历
for(int j = result.length - 2;j >= 0;j--){
if(ratings[j] > ratings[j + 1] && result[j] <= result[j + 1]){
result[j] = result[j + 1] + 1;
}
}
// 计算总糖果数的和
for(int i = 0;i < result.length;i++){
count += result[i];
}
return count;
}
}