贪心算法的核心思想,程序的每一步都是向目标结果靠拢,每一步都是最优解 最终结果就是最优解。
//老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
//
// 你需要按照以下要求,帮助老师给这些孩子分发糖果:
//
//
// 每个孩子至少分配到 1 个糖果。
// 相邻的孩子中,评分高的孩子必须获得更多的糖果。
//
//
// 那么这样下来,老师至少需要准备多少颗糖果呢?
//
// 示例 1:
//
// 输入: [1,0,2]
//输出: 5
//解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
//
//
// 示例 2:
//
// 输入: [1,2,2]
//输出: 4
//解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。
// 第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
// Related Topics 贪心算法
// 👍 318 👎 0
import java.util.Map;
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int candy(int[] ratings) {
int[] sun= new int[ratings.length];//记录每个孩子所需要的糖果数量
//保证每个孩子至少分配到 1 个糖果。
for (int i = 0; i < sun.length; i++) {
sun[i]=1;
}
//相邻的孩子中,评分高的孩子必须获得更多的糖果。cur表示当前孩子,next 表示下一个孩子
int cur = 0;
int next = 0;
Map<Integer, Integer> map = new HashMap<>();
//从左到右遍历 此次遍历保证右边的数比左边的数大的情况糖果数
for (int i = 0; i < ratings.length - 1; i++) {
cur = ratings[i];
next = ratings[i + 1];
if (cur == next) {
continue;//相邻孩子评分一样不用处理
}
if (cur < next) {//如果后面的评分高,题目中要求(至少需要准备多少颗糖果呢)至少我们只需要后边孩子比前面的多一个就行
if(sun[i]==sun[i + 1]){//初始化时候都是1
sun[i+1] = sun[i+1] + 1;
}
if(sun[i]>sun[i + 1]){//保证右的孩子糖果个数比左边多
sun[i + 1]=sun[i]+1;
}
}
}
//从右到左遍历此次遍历保证左边的数比右边的数大的情况时的糖果数
for (int i = ratings.length-1; i >0; i--) {
cur = ratings[i];
next = ratings[i - 1];
//此处逻辑和上面一样,保证左边的孩子糖果个数比右边多
if (cur == next) {
continue;
}
if (cur < next) {
if(sun[i]==sun[i - 1]){
sun[i-1] = sun[i - 1] + 1;
}
if(sun[i]>sun[i - 1]){
sun[i-1]=sun[i]+1;
}
}
}
int sum=0;
System.out.println(sun);
for (int i = 0; i < sun.length; i++) {
sum=sum+sun[i];
}
return sum;
}
}
//leetcode submit region end(Prohibit modification and deletion)