题目描述
135. 分发糖果
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/candy
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
要求相邻的且评分高的孩子获得更多的糖果,遍历ratings[]数组可选择左右各遍历一次。
对于左遍历,当前孩子i需要满足ratings[i] > ratings[i - 1]时,则孩子i的糖果数量比孩子i-1多;
对于右遍历,当前孩子i需要满足ratings[i] > ratings[i + 1]时,则孩子i的糖果数量比孩子i+1多;
最后根据左右遍历的结果依次比较,取最大值,即是当前孩子i的最大糖果数量。
ACM代码–Java
import java.util.*;
public class test13 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
System.out.println("请输入孩子个数:");
int n = sc.nextInt();
int[] ratings = new int[n];
for (int i = 0; i < n; i++) {
ratings[i] = sc.nextInt();
System.out.println(ratings[i] + " ");
}
System.out.println();
System.out.println("最少糖果数目为: " + candy(ratings));
}
public static int candy(int[] ratings) {
int n = ratings.length;
int[] left = new int[n];
for (int i = 0; i < n; i++) {
if (i > 0 && ratings[i] > ratings[i - 1]) {
left[i] = left[i - 1] + 1;
} else {
left[i] = 1;
}
}
int right = 0, res = 0;
for (int i = n - 1; i >= 0; i--) {
if (i < n - 1 && ratings[i] > ratings[i + 1]) {
right++;
} else {
right = 1;
}
res += Math.max(left[i], right);
}
return res;
}
}